bnsreenu / python_for_microscopists

https://www.youtube.com/channel/UC34rW-HtPJulxr5wp2Xa04w?sub_confirmation=1
MIT License
3.89k stars 2.38k forks source link

index 4 is out of bounds for axis 1 with size 3 #37

Open DebarunChakraborty opened 2 years ago

DebarunChakraborty commented 2 years ago

Sir, I m trying to train the model for the segmentation, but I getting an error in the Index.

"index 4 is out of bounds for axis 1 with size 3" A similar issue has been already raised but you didn't reply. Please sir it will be a great help if you can address this issue.

Hoping for your positive response earliest.

rahuldevyadav commented 2 years ago

Hi @DebarunChakraborty if u are using new version of dataset then there number of class will be 5 (because road is also labelled in new dataset) so use n_classes=5

DebarunChakraborty commented 2 years ago

Hi @rahuldevyadav Thank you