diff --git a/README.md b/README.md index f32ca20e35..dea03e097c 100644 --- a/README.md +++ b/README.md @@ -57,37 +57,38 @@ Results and models are available in the [model zoo](docs/model_zoo.md). Supported backbones: -- [x] ResNet -- [x] ResNeXt -- [x] [HRNet](configs/hrnet/README.md) -- [x] [ResNeSt](configs/resnest/README.md) -- [x] [MobileNetV2](configs/mobilenet_v2/README.md) -- [x] [MobileNetV3](configs/mobilenet_v3/README.md) +- [x] ResNet (CVPR'2016) +- [x] ResNeXt (CVPR'2017) +- [x] [HRNet (CVPR'2019)](configs/hrnet/README.md) +- [x] [ResNeSt (ArXiv'2020)](configs/resnest/README.md) +- [x] [MobileNetV2 (CVPR'2018)](configs/mobilenet_v2/README.md) +- [x] [MobileNetV3 (ICCV'2019)](configs/mobilenet_v3/README.md) Supported methods: -- [x] [FCN](configs/fcn) -- [x] [PSPNet](configs/pspnet) -- [x] [DeepLabV3](configs/deeplabv3) -- [x] [PSANet](configs/psanet) -- [x] [DeepLabV3+](configs/deeplabv3plus) -- [x] [UPerNet](configs/upernet) -- [x] [NonLocal Net](configs/nonlocal_net) -- [x] [EncNet](configs/encnet) -- [x] [CCNet](configs/ccnet) -- [x] [DANet](configs/danet) -- [x] [APCNet](configs/apcnet) -- [x] [GCNet](configs/gcnet) -- [x] [DMNet](configs/dmnet) -- [x] [ANN](configs/ann) -- [x] [OCRNet](configs/ocrnet) -- [x] [Fast-SCNN](configs/fastscnn) -- [x] [Semantic FPN](configs/sem_fpn) -- [x] [PointRend](configs/point_rend) -- [x] [EMANet](configs/emanet) -- [x] [DNLNet](configs/dnlnet) -- [x] [CGNet](configs/cgnet) -- [x] [Mixed Precision (FP16) Training](configs/fp16/README.md) +- [x] [FCN (CVPR'2015/TPAMI'2017)](configs/fcn) +- [x] [UNet (MICCAI'2016/Nat. Methods'2019)](configs/unet) +- [x] [PSPNet (CVPR'2017)](configs/pspnet) +- [x] [DeepLabV3 (CVPR'2017)](configs/deeplabv3) +- [x] [Mixed Precision (FP16) Training (ArXiv'2017)](configs/fp16/README.md) +- [x] [PSANet (ECCV'2018)](configs/psanet) +- [x] [DeepLabV3+ (CVPR'2018)](configs/deeplabv3plus) +- [x] [UPerNet (ECCV'2018)](configs/upernet) +- [x] [NonLocal Net (CVPR'2018)](configs/nonlocal_net) +- [x] [EncNet (CVPR'2018)](configs/encnet) +- [x] [Semantic FPN (CVPR'2019)](configs/sem_fpn) +- [x] [DANet (CVPR'2019)](configs/danet) +- [x] [APCNet (CVPR'2019)](configs/apcnet) +- [x] [EMANet (ICCV'2019)](configs/emanet) +- [x] [CCNet (ICCV'2019)](configs/ccnet) +- [x] [DMNet (ICCV'2019)](configs/dmnet) +- [x] [ANN (ICCV'2019)](configs/ann) +- [x] [GCNet (ICCVW'2019/TPAMI'2020)](configs/gcnet) +- [x] [Fast-SCNN (ArXiv'2019)](configs/fastscnn) +- [x] [OCRNet (ECCV'2020)](configs/ocrnet) +- [x] [DNLNet (ECCV'2020)](configs/dnlnet) +- [x] [PointRend (CVPR'2020)](configs/point_rend) +- [x] [CGNet (TIP'2020)](configs/cgnet) ## Installation diff --git a/README_zh-CN.md b/README_zh-CN.md index df7439485d..a99c24968a 100644 --- a/README_zh-CN.md +++ b/README_zh-CN.md @@ -56,37 +56,37 @@ MMSegmentation 是一个基于 PyTorch 的语义分割开源工具箱。它是 O 已支持的骨干网络： -- [x] ResNet -- [x] ResNeXt -- [x] [HRNet](configs/hrnet/README.md) -- [x] [ResNeSt](configs/resnest/README.md) -- [x] [MobileNetV2](configs/mobilenet_v2/README.md) -- [x] [MobileNetV3](configs/mobilenet_v3/README.md) +- [x] ResNet (CVPR'2016) +- [x] ResNeXt (CVPR'2017) +- [x] [HRNet (CVPR'2019)](configs/hrnet/README.md) +- [x] [ResNeSt (ArXiv'2020)](configs/resnest/README.md) +- [x] [MobileNetV2 (CVPR'2018)](configs/mobilenet_v2/README.md) +- [x] [MobileNetV3 (ICCV'2019)](configs/mobilenet_v3/README.md) 已支持的算法： -- [x] [FCN](configs/fcn) -- [x] [PSPNet](configs/pspnet) -- [x] [DeepLabV3](configs/deeplabv3) -- [x] [PSANet](configs/psanet) -- [x] [DeepLabV3+](configs/deeplabv3plus) -- [x] [UPerNet](configs/upernet) -- [x] [NonLocal Net](configs/nonlocal_net) -- [x] [EncNet](configs/encnet) -- [x] [CCNet](configs/ccnet) -- [x] [DANet](configs/danet) -- [x] [APCNet](configs/apcnet) -- [x] [GCNet](configs/gcnet) -- [x] [DMNet](configs/dmnet) -- [x] [ANN](configs/ann) -- [x] [OCRNet](configs/ocrnet) -- [x] [Fast-SCNN](configs/fastscnn) -- [x] [Semantic FPN](configs/sem_fpn) -- [x] [PointRend](configs/point_rend) -- [x] [EMANet](configs/emanet) -- [x] [DNLNet](configs/dnlnet) -- [x] [CGNet](configs/cgnet) -- [x] [Mixed Precision (FP16) Training](configs/fp16/README.md) +- [x] [FCN (CVPR'2015/TPAMI'2017)](configs/fcn) +- [x] [PSPNet (CVPR'2017)](configs/pspnet) +- [x] [DeepLabV3 (CVPR'2017)](configs/deeplabv3) +- [x] [Mixed Precision (FP16) Training (ArXiv'2017)](configs/fp16/README.md) +- [x] [PSANet (ECCV'2018)](configs/psanet) +- [x] [DeepLabV3+ (CVPR'2018)](configs/deeplabv3plus) +- [x] [UPerNet (ECCV'2018)](configs/upernet) +- [x] [NonLocal Net (CVPR'2018)](configs/nonlocal_net) +- [x] [EncNet (CVPR'2018)](configs/encnet) +- [x] [Semantic FPN (CVPR'2019)](configs/sem_fpn) +- [x] [DANet (CVPR'2019)](configs/danet) +- [x] [APCNet (CVPR'2019)](configs/apcnet) +- [x] [EMANet (ICCV'2019)](configs/emanet) +- [x] [CCNet (ICCV'2019)](configs/ccnet) +- [x] [DMNet (ICCV'2019)](configs/dmnet) +- [x] [ANN (ICCV'2019)](configs/ann) +- [x] [GCNet (ICCVW'2019/TPAMI'2020)](configs/gcnet) +- [x] [Fast-SCNN (ArXiv'2019)](configs/fastscnn) +- [x] [OCRNet (ECCV'2020)](configs/ocrnet) +- [x] [DNLNet (ECCV'2020)](configs/dnlnet) +- [x] [PointRend (CVPR'2020)](configs/point_rend) +- [x] [CGNet (TIP'2020)](configs/cgnet) ## 安装 diff --git a/configs/cgnet/README.md b/configs/cgnet/README.md index 4859492ebf..0acdfb7138 100644 --- a/configs/cgnet/README.md +++ b/configs/cgnet/README.md @@ -5,11 +5,14 @@ [ALGORITHM] ```latext -@article{wu2018cgnet, - title={CGNet: A Light-weight Context Guided Network for Semantic Segmentation}, - author={Wu, Tianyi and Tang, Sheng and Zhang, Rui and Zhang, Yongdong}, - journal={arXiv preprint arXiv:1811.08201}, - year={2018} +@article{wu2020cgnet, + title={Cgnet: A light-weight context guided network for semantic segmentation}, + author={Wu, Tianyi and Tang, Sheng and Zhang, Rui and Cao, Juan and Zhang, Yongdong}, + journal={IEEE Transactions on Image Processing}, + volume={30}, + pages={1169--1179}, + year={2020}, + publisher={IEEE} } ``` diff --git a/configs/point_rend/README.md b/configs/point_rend/README.md index af429665e8..1d00f0ecf1 100644 --- a/configs/point_rend/README.md +++ b/configs/point_rend/README.md @@ -5,13 +5,12 @@ [ALGORITHM] ``` -@misc{alex2019pointrend, - title={PointRend: Image Segmentation as Rendering}, - author={Alexander Kirillov and Yuxin Wu and Kaiming He and Ross Girshick}, - year={2019}, - eprint={1912.08193}, - archivePrefix={arXiv}, - primaryClass={cs.CV} +@inproceedings{kirillov2020pointrend, + title={Pointrend: Image segmentation as rendering}, + author={Kirillov, Alexander and Wu, Yuxin and He, Kaiming and Girshick, Ross}, + booktitle={Proceedings of the IEEE/CVF conference on computer vision and pattern recognition}, + pages={9799--9808}, + year={2020} } ``` diff --git a/configs/unet/README.md b/configs/unet/README.md index 1059815af4..c3d0484b9c 100644 --- a/configs/unet/README.md +++ b/configs/unet/README.md @@ -35,16 +35,16 @@ ### CHASE_DB1 -| Backbone | Head | Image Size | Crop Size | Stride | Lr schd | Mem (GB) | Inf time (fps) | Dice | download | -| ----------- | --------- | ---------- | --------- | -----: | ------- | -------- | -------------: | ----: | ------------------------------------------------------------------------------------------------------------------------------ | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | +| Backbone | Head | Image Size | Crop Size | Stride | Lr schd | Mem (GB) | Inf time (fps) | Dice | config | download | +| ----------- | --------- | ---------- | --------- | -----: | ------- | -------- | -------------: | ----: | -------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | | UNet-S5-D16 | FCN | 960x999 | 128x128 | 85x85 | 40000 | 0.968 | - | 80.24 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/fcn_unet_s5-d16_128x128_40k_chase_db1.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/unet/fcn_unet_s5-d16_128x128_40k_chase_db1/fcn_unet_s5-d16_128x128_40k_chase_db1_20201223_191051-95852f45.pth) | [log](https://download.openmmlab.com/mmsegmentation/v0.5/unet/fcn_unet_s5-d16_128x128_40k_chase_db1/fcn_unet_s5-d16_128x128_40k_chase_db1-20201223_191051.log.json) | | UNet-S5-D16 | PSPNet | 960x999 | 128x128 | 85x85 | 40000 | 0.982 | - | 80.36 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/pspnet_unet_s5-d16_128x128_40k_chase_db1.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/unet/pspnet_unet_s5-d16_128x128_40k_chase_db1/pspnet_unet_s5-d16_128x128_40k_chase_db1_20201227_181818-68d4e609.pth) | [log](https://download.openmmlab.com/mmsegmentation/v0.5/unet/pspnet_unet_s5-d16_128x128_40k_chase_db1/pspnet_unet_s5-d16_128x128_40k_chase_db1-20201227_181818.log.json) | | UNet-S5-D16 | DeepLabV3 | 960x999 | 128x128 | 85x85 | 40000 | 0.999 | - | 80.47 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/unet/deeplabv3_unet_s5-d16_128x128_40k_chase_db1/deeplabv3_unet_s5-d16_128x128_40k_chase_db1_20201226_094047-4c5aefa3.pth) | [log](https://download.openmmlab.com/mmsegmentation/v0.5/unet/deeplabv3_unet_s5-d16_128x128_40k_chase_db1/deeplabv3_unet_s5-d16_128x128_40k_chase_db1-20201226_094047.log.json) | ### HRF -| Backbone | Head | Image Size | Crop Size | Stride | Lr schd | Mem (GB) | Inf time (fps) | Dice | download | -| ----------- | --------- | ---------- | --------- | ------: | ------- | -------- | -------------: | ----: | ------------------------------------------------------------------------------------------------------------------------ | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | +| Backbone | Head | Image Size | Crop Size | Stride | Lr schd | Mem (GB) | Inf time (fps) | Dice | config | download | +| ----------- | --------- | ---------- | --------- | -----: | ------- | -------- | -------------: | ----: | -------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | | UNet-S5-D16 | FCN | 2336x3504 | 256x256 | 170x170 | 40000 | 2.525 | - | 79.45 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/fcn_unet_s5-d16_256x256_40k_hrf.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/unet/fcn_unet_s5-d16_256x256_40k_hrf/fcn_unet_s5-d16_256x256_40k_hrf_20201223_173724-df3ec8c4.pth) | [log](https://download.openmmlab.com/mmsegmentation/v0.5/unet/fcn_unet_s5-d16_256x256_40k_hrf/fcn_unet_s5-d16_256x256_40k_hrf-20201223_173724.log.json) | | UNet-S5-D16 | PSPNet | 2336x3504 | 256x256 | 170x170 | 40000 | 2.588 | - | 80.07 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/pspnet_unet_s5-d16_256x256_40k_hrf.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/unet/pspnet_unet_s5-d16_256x256_40k_hrf/pspnet_unet_s5-d16_256x256_40k_hrf_20201227_181818-fdb7e29b.pth) | [log](https://download.openmmlab.com/mmsegmentation/v0.5/unet/pspnet_unet_s5-d16_256x256_40k_hrf/pspnet_unet_s5-d16_256x256_40k_hrf-20201227_181818.log.json) | | UNet-S5-D16 | DeepLabV3 | 2336x3504 | 256x256 | 170x170 | 40000 | 2.604 | - | 80.21 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/deeplabv3_unet_s5-d16_256x256_40k_hrf.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/unet/deeplabv3_unet_s5-d16_256x256_40k_hrf/deeplabv3_unet_s5-d16_256x256_40k_hrf_20201226_094047-3a1fdf85.pth) | [log](https://download.openmmlab.com/mmsegmentation/v0.5/unet/deeplabv3_unet_s5-d16_256x256_40k_hrf/deeplabv3_unet_s5-d16_256x256_40k_hrf-20201226_094047.log.json) |