CSAILVision / CSAILVision/semantic-segmentation-pytorch

Trained with custom dataset model results

Open
#279 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
5.1k
Forks
1.1k
PR merge metrics
No merged PRs in 30d

Description

I prepared custom dataset that 7000 images has only floor class and added to ade20k dataset. Now i have dataset with 28000 images. I changed floor annotations color with #040404 in my dataset, because floor class had that color in ADE20K dataset. So my floor annotation looks like this;
![0101 jpg___fuse](https://user-images.githubusercontent.com/7453897/184089710-815c440d-c3f7-4771-9432-2b200da77cf3.png)
(#040404 color is very close to black. If you look carefully you will see the floor annotation)

and original images;
![0101](https://user-images.githubusercontent.com/7453897/184089839-cf465eea-8f90-49b2-ae55-5779f1c97919.jpg)

It seems everything normal my dataset. Then i changed start_epoch and num_epoch values in config yaml file,
`num_epoch: 23
start_epoch: 20
epoch_iters: 5000` .
Training process done with successfully and i have encoder and decoder model that names are encoder_epoch_23.pth, decoder_epoch_23.pth. Everything is seems normal here as well

I got results using theses models but result was not as expected.
I got this result when i download model from here [decoder_epoch_20.pth](http://sceneparsing.csail.mit.edu/model/pytorch/ade20k-resnet50dilated-ppm_deepsup/decoder_epoch_20.pth)
![f89bce9bcb394a8a8aa785dfb847bf4a](https://user-images.githubusercontent.com/7453897/184092798-554d28d1-1a9e-41e0-be8b-681fdab7e9b7.png)

And i got this result when using i trained model;

![03046ef6e5ac431e8e33002854eebb05](https://user-images.githubusercontent.com/7453897/184092967-8b537963-5799-4006-a3e9-1dadc4d9be3b.png)

Results seem to be getting worse. What could i be doing wrong?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.