谢昕辰 725d5aa002 [Feature] support mim (#549)
* dice loss

* format code, add docstring and calculate denominator without valid_mask

* minor change

* restore

* add metafile

* add manifest.in and add config at setup.py

* add requirements

* modify manifest

* modify manifest

* Update MANIFEST.in

* add metafile

* add metadata

* fix typo

* Update metafile.yml

* Update metafile.yml

* minor change

* Update metafile.yml

* add subfix

* fix mmshow

* add more  metafile

* add config to model_zoo

* fix bug

* Update mminstall.txt

* [fix] Add models

* [Fix] Add collections

* [fix] Modify collection name

* [Fix] Set datasets to unet metafile

* [Fix] Modify collection names

* complement inference time
2021-05-31 15:07:24 -07:00

64 lines
1.9 KiB
YAML

Collections:
- Name: FPN
Metadata:
Training Data:
- Cityscapes
- Pascal VOC 2012 + Aug
- ADE20K
Models:
- Name: fpn_r50_512x1024_80k_cityscapes
In Collection: FPN
Metadata:
inference time (fps): 13.54
Results:
- Task: Semantic Segmentation
Dataset: Cityscapes
Metrics:
mIoU: 74.52
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/sem_fpn/fpn_r50_512x1024_80k_cityscapes/fpn_r50_512x1024_80k_cityscapes_20200717_021437-94018a0d.pth
Config: configs/fpn/fpn_r50_512x1024_80k_cityscapes.py
- Name: fpn_r101_512x1024_80k_cityscapes
In Collection: FPN
Metadata:
inference time (fps): 10.29
Results:
- Task: Semantic Segmentation
Dataset: Cityscapes
Metrics:
mIoU: 75.80
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/sem_fpn/fpn_r101_512x1024_80k_cityscapes/fpn_r101_512x1024_80k_cityscapes_20200717_012416-c5800d4c.pth
Config: configs/fpn/fpn_r101_512x1024_80k_cityscapes.py
- Name: fpn_r50_512x512_160k_ade20k
In Collection: FPN
Metadata:
inference time (fps): 55.77
Results:
- Task: Semantic Segmentation
Dataset: ADE20K
Metrics:
mIoU: 37.49
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/sem_fpn/fpn_r50_512x512_160k_ade20k/fpn_r50_512x512_160k_ade20k_20200718_131734-5b5a6ab9.pth
Config: configs/fpn/fpn_r50_512x512_160k_ade20k.py
- Name: fpn_r101_512x512_160k_ade20k
In Collection: FPN
Metadata:
inference time (fps): 40.58
Results:
- Task: Semantic Segmentation
Dataset: ADE20K
Metrics:
mIoU: 39.35
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/sem_fpn/fpn_r101_512x512_160k_ade20k/fpn_r101_512x512_160k_ade20k_20200718_131734-306b5004.pth
Config: configs/fpn/fpn_r101_512x512_160k_ade20k.py