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.