/semantic-segmentation-projects

Primary LanguagePython

Segmentation Projects

实现了一些常用的语义分割算法，可以支持VOC以及自定义的数据集训练

环境依赖

pytorch-1.6 PIL Opencv thop

支持的算法

  1. deeplabv3++
  2. enet
  3. espnet
  4. icnet
  5. pspnet
  6. gscnn
  7. decouplesegnet
  8. danet
  9. cpnet
  10. ocrnet
  11. bisenetv1
  12. bisenetv2
  13. sfsegnet

Training

参数可以在opts.py中配置，比如一些关键参数，如--model,--batch_size, --lr, --dataset, --data_root等等。
python main.py --model bisenetv1 --gpu_id 0 --year 2012_aug --crop_val --lr 0.01 --crop_size 640 --batch_size 16