bodokaiser / piwise

Pixel-wise segmentation on VOC2012 dataset using pytorch.
BSD 3-Clause "New" or "Revised" License
385 stars 86 forks source link

update models #1

Closed bodokaiser closed 7 years ago

bodokaiser commented 7 years ago

We want to implement following models (ordered after priority):

  1. FCN
  2. SqueezeNet
  3. SegNet
  4. PSPNet
  5. RefineNet

Removed models (because too big, too complicated or not much information):

bodokaiser commented 7 years ago

FCN

We use the this VGG-8 implementation which can be viewed as graph.

SegNet

PSPNet

https://github.com/hszhao/PSPNet/blob/master/evaluation/prototxt/pspnet101_VOC2012_473.prototxt

RefineNet

https://github.com/guosheng/refinenet/tree/master/net_graphs

SqueezeNet

https://github.com/DeepScale/SqueezeNet/blob/master/SqueezeNet_v1.1/train_val.prototxt