search

meetps / pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch
https://meetshah.dev/semantic-segmentation/deep-learning/pytorch/visdom/2017/06/01/semantic-segmentation-over-the-years.html
MIT License
3.39k stars 796 forks source link

Classweighting of SegNet? #202

Open qiaobaiyisheng opened 5 years ago

qiaobaiyisheng commented 5 years ago

If I use the SenNet, do I need to compute the classweighting mentioned in the paper?

Qirui-Y commented 5 years ago

Hello, are you solved problem ? Can you share your config file?

qiaobaiyisheng commented 5 years ago

model: arch: segnet data: dataset: camvid train_split: train val_split: val img_rows: 360 img_cols: 480 path: /your/path/pytorch-semseg-master/data/CamVid/ training: train_iters: 150000 batch_size: 8 val_interval: 100 n_workers: 16 print_interval: 50 optimizer: name: 'sgd' lr: 1.0e-10 weight_decay: 0.0005 momentum: 0.99 loss: name: 'cross_entropy' size_average: False lr_schedule: resume: segnet_camvid_model.pkl

I didn't found where to compute the classweighting in this code, so directly run the program with the config file above. The result on CamVid is about 0.58(mIOU), which is close to the experimental result in the proposed SegNet paper.

Qirui-Y commented 5 years ago

OK，thank you very much,I am try it!