deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard

Published 20 hours ago • verified publisher icon WZMIAOMIAO

Unet Multiclass training process

Open sharifsagar80 opened this issue 2 years ago • 0 comments

Hi, Your repository is very helpful. I was trying to run the Unet model for the multiclass dataset, where every class is labeled with different colors to generate masks. But when I tried to run the code, it was not working. I have changed the num classes to the class number of the dataset, but it shows an error from the "train one epoch class." Can you share any process to enable multiclass training?

sharifsagar80 avatar Nov 20 '22 20:11 sharifsagar80