search

dmlc / gluon-cv

Gluon CV Toolkit
http://gluon-cv.mxnet.io
Apache License 2.0
5.83k stars 1.22k forks source link

Is there any plans to release more segmentation models in near future? #775

Closed shadyatscu closed 3 years ago

shadyatscu commented 5 years ago

Different from other deep learning frameworks, mxnet is very convenient for beginners as the official model zoo has integrated with SOTA algorithms. In the segmentation part, some popular algorithms are missing, eg unet series, linknet and so on... Is there ant plans to add those SOTA algorithms in the near future or any roadmaps?

Mut1nyJD commented 5 years ago

I am looking at implementing FastFCN and ShelfNet I am happy to contribute once I got it working.

zhanghang1989 commented 5 years ago

DeepLabV3Plus using Xception backbone was just added https://github.com/dmlc/gluon-cv/blob/master/gluoncv/model_zoo/deeplabv3_plus.py#L243 I will provide pretrained model weights soon.

OliverColeman commented 5 years ago

While we're waiting for pretrained models (although it looks like there are some for COCO in the code), can you confirm whether you train this in two stages:

  1. not dilated with output stride 32, large batch size
  2. dilated with output stride 8, small batch size, with batchnorm params fixed?
OliverColeman commented 5 years ago

Actually, looks like pretrained model params not actually available yet.

github-actions[bot] commented 3 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.