kris-singh / Deep-Unsupervised-Saliency-Detection

36 stars 8 forks source link

Model does not train well with noise module #4

Open GerardWalsh opened 3 years ago

GerardWalsh commented 3 years ago

Hi, I have a small dataset (300 training images). The model performs ok during round 1 of training, but when introducing the noise module, the prediction loss returns to where it started in round 1 and does not improve. The noise loss does not decrease over time either. Any ideas? I'm using two sources of noisy ground truth.