search

lattice-ai / DeepLabV3-Plus

Tensorflow 2.3.0 implementation of DeepLabV3-Plus
https://keras.io/examples/vision/deeplabv3_plus/
101 stars 23 forks source link

5 classes #48

Open faxirabd opened 1 year ago

faxirabd commented 1 year ago

Hello, I just tried using deeplabv3+ code with only 5 classes instead of 20. The training goes gut, however the evaluation loss goes to nan (infinity) and testing is the same. Is this model works only with 20 classes?

Barthas255 commented 1 year ago

I used it for only 2 classes and it seems to work fine