Hi songyuc,
Thanks for you feedback! Do you use multiclass segmentation by chance? If you could check the reply I gave to ptrblck and if you could give me a feedback on how you deal with the mask loading and manipulation would be great. Sorry I’m quite new to pytorch so I might ask some very basic questions.
Best