search

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images
GNU General Public License v3.0
8.92k stars 2.45k forks source link

Multiclass dice score #440

Open leosmaia21 opened 1 year ago

leosmaia21 commented 1 year ago

Hello I am pretrainig the model on 20K images from cocodataset, and right after the first epoch i have a dice score of 0.91, but i tested the model and it doest work well, the mask returned all with 0's.

Thanks image

ahmetfurkaann commented 1 year ago

I have the same error. what is the problem with this