weights
Opened this issue · 0 comments
hadi1994shokati commented
Hello. Thank you for sharing the code.
I want to use your code for the segmentation of 2 classes using resnet50. When I run the code, it uploads the best.pth file, but this file has been prepared for 16 classes and gets an error for the segmentation of 2 classes. How can I apply that file for 2 classes? Thanks