nauyan / Segmentation

Contains code for Semantic Segmentation of MoNuSeg 2018 challenge.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

UNTMOD issue

zzw0812 opened this issue · comments

when I run the Test.py to test the UNTMOD model, I got this error: "ValueError: Unknown layer: BilinearUpSampling2D" , I want to know how to solve this problem? Thanks!

Have you installed all the required libraries present in the requirments.txt file?

Have you installed all the required libraries present in the requirments.txt file?

yes, I am sure. By the way, you didn't add the UNTMOD model in the Test.py, so I add it like UNET style in the Test.py, just like this:
if config['Model']=="UNETMOD":
model = keras.models.load_model("./Results/weights/" + config['Model'] + "/" +config['Model']+"-Best.h5", compile=False)

I have just updated the repo and modified the Test.py file but you would have to add a custom layer of BilinearUpSampling2D while loading weights.

I have just updated the repo and modified the Test.py file but you would have to add a custom layer of BilinearUpSampling2D while loading weights.
Thanks for your reply, yes, I have add a custom layer by myself, the code just like this:, am right?

Yes that is fine.