2825efe378
* add DPT head * [fix] fix init error * use mmcv function * delete code * remove transpose clas * support NLC output shape * Delete post_process_layer.py * add unittest and docstring * rename variables * fix project error and add unittest * match dpt weights * add configs * fix vit pos_embed bug and dpt feature fusion bug * match vit output * fix gelu * minor change * update unitest * fix configs error * inference test * remove auxilary * use local pretrain * update training results * update yml * update fps and memory test * update doc * update readme * add yml * update doc * remove with_cp * update config * update docstring * remove dpt-l * add init_cfg and modify readme.md * Update dpt_vit-b16.py * zh-n README * use constructor instead of build function * prevent tensor being modified by ConvModule * fix unittest Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
29 lines
714 B
YAML
29 lines
714 B
YAML
Collections:
|
|
- Metadata:
|
|
Training Data:
|
|
- ADE20K
|
|
Name: dpt
|
|
Models:
|
|
- Config: configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py
|
|
In Collection: dpt
|
|
Metadata:
|
|
backbone: ViT-B
|
|
crop size: (512,512)
|
|
inference time (ms/im):
|
|
- backend: PyTorch
|
|
batch size: 1
|
|
hardware: V100
|
|
mode: FP32
|
|
resolution: (512,512)
|
|
value: 96.06
|
|
lr schd: 160000
|
|
memory (GB): 8.09
|
|
Name: dpt_vit-b16_512x512_160k_ade20k
|
|
Results:
|
|
Dataset: ADE20K
|
|
Metrics:
|
|
mIoU: 46.97
|
|
mIoU(ms+flip): 48.34
|
|
Task: Semantic Segmentation
|
|
Weights: https://download.openmmlab.com/mmsegmentation/v0.5/dpt/dpt_vit-b16_512x512_160k_ade20k/dpt_vit-b16_512x512_160k_ade20k-db31cf52.pth
|