Hugging Face's logo

KieDani
/
SegformerPlusPlus

Image Segmentation
English
Model card Files
xet
Community
KieDani commited on
Commit
7b37eed
·
verified ·
1 Parent(s): 2c95d7e

Removed all config stuff from readme

Browse files
Files changed (1) hide show
  1. README.md +33 -33
README.md CHANGED
@@ -28,51 +28,51 @@ Memory refers to the VRAM requirements during the training process.
28
 
29
  The weights of the Segformer (Original) model were used to get the inference results.
30
 
31
- | Method | mIoU | Speed-Up | config | download |
32
- |-----------------------------------|------:|---------:|--------------------------------------------------------------------------------------------|----------------------------------------------------------------|
33
- | Segformer (Original) | 82.39 | - | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-default.py) | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
34
- | Segformer++<sub>HQ</sub> (ours) | 82.31 | 1.61 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-bsm-hq.py) | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
35
- | Segformer++<sub>fast</sub> (ours) | 82.04 | 1.94 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-bsm-fast.py) | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
36
- | Segformer++<sub>2x2</sub> (ours) | 81.96 | 1.90 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-n2d-2x2.py) | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
37
- | Segformer (Downsampling) | 77.31 | 6.51 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-downsample.py) | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
38
 
39
  ### Training on Cityscapes (MiT-B5)
40
 
41
- | Method | mIoU | Speed-Up | Memory (GB) | config | download |
42
- |-----------------------------------|------:|---------:|-------------|--------------------------------------------------------------------------------------------|-----------------------------------------------------------------|
43
- | Segformer (Original) | 82.39 | - | 48.3 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-default.py) | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
44
- | Segformer++<sub>HQ</sub> (ours) | 82.19 | 1.40 | 34.0 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-bsm-hq.py) | [model](https://mediastore.rz.uni-augsburg.de/get/i8fY8uXJrV/ ) |
45
- | Segformer++<sub>fast</sub> (ours) | 81.77 | 1.55 | 30.5 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-bsm-fast.py) | [model](https://mediastore.rz.uni-augsburg.de/get/cmG974iAxt/ ) |
46
- | Segformer++<sub>2x2</sub> (ours) | 82.38 | 1.63 | 31.1 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-n2d-2x2.py) | [model](https://mediastore.rz.uni-augsburg.de/get/p0uMKbw531/) |
47
- | Segformer (Downsampling) | 79.24 | 2.95 | 10.0 | [config](segmentation_configs/cityscapes/B5/segformer-cityscapes-b5-downsample.py) | [model](https://mediastore.rz.uni-augsburg.de/get/73zkKSO21t/) |
48
 
49
  ### Training on ADE20K (640x640) (MiT-B5)
50
 
51
- | Method | mIoU | Speed-Up | Memory (GB) | config | download |
52
- |-----------------------------------|------:|---------:|------------:|---------------------------------------------------------------------------------------|----------------------------------------------------------------|
53
- | Segformer (Original) | 49.72 | - | 33.7 | [config](segmentation_configs/ade20k/B5/segformer-ade20k640-b5-default.py) | [model](https://mediastore.rz.uni-augsburg.de/get/nKEjUHNAfK/) |
54
- | Segformer++<sub>HQ</sub> (ours) | 49.77 | 1.15 | 29.2 | [config](segmentation_configs/ade20k/B5/segformer-ade20k640-b5-bsm-hq.py) | [model](https://mediastore.rz.uni-augsburg.de/get/Odyie8usgj/) |
55
- | Segformer++<sub>fast</sub> (ours) | 49.10 | 1.20 | 28.0 | [config](segmentation_configs/ade20k/B5/segformer-ade20k640-b5-bsm-fast.py) | [model](https://mediastore.rz.uni-augsburg.de/get/K0IGkx4O2s/) |
56
- | Segformer++<sub>2x2</sub> (ours) | 49.35 | 1.26 | 27.2 | [config](segmentation_configs/ade20k/B5/segformer-ade20k640-b5-n2d-2x2.py) | [model](https://mediastore.rz.uni-augsburg.de/get/w5_Pxx4Q5C/) |
57
- | Segformer (Downsampling) | 46.71 | 1.89 | 12.4 | [config](segmentation_configs/ade20k/B5/segformer-ade20k640-b5-downsample.py) | [model](https://mediastore.rz.uni-augsburg.de/get/dFVvZQL6iL/) |
58
 
59
  ### Training on JBD
60
 
61
- | Method | PCK@0.1 | PCK@0.05 | Speed-Up | Memory (GB) | config | download |
62
- |-----------------------------------|--------:|---------:|---------:|------------:|---------------------------------------------------------------------|----------------------------------------------------------------|
63
- | Segformer (Original) | 95.20 | 90.65 | - | 40.0 | [config](hpe_configs/jbd/B5/segformer-jump-b5-default.py) | [model](https://mediastore.rz.uni-augsburg.de/get/psolrWXLLp/) |
64
- | Segformer++<sub>HQ</sub> (ours) | 95.18 | 90.51 | 1.19 | 36.0 | [config](hpe_configs/jbd/B5/segformer-jump-b5-bsm-hq.py) | [model](https://mediastore.rz.uni-augsburg.de/get/jx1eyecMLF/) |
65
- | Segformer++<sub>fast</sub> (ours) | 94.58 | 89.87 | 1.25 | 34.6 | [config](hpe_configs/jbd/B5/segformer-jump-b5-bsm-fast.py) | [model](https://mediastore.rz.uni-augsburg.de/get/K0IGkx4O2s/) |
66
- | Segformer++<sub>2x2</sub> (ours) | 95.17 | 90.16 | 1.27 | 33.4 | [config](hpe_configs/jbd/B5/segformer-jump-b5-n2d-2x2.py) | [model](https://mediastore.rz.uni-augsburg.de/get/HumKbSB1vI/) |
67
 
68
  ### Training on MS COCO
69
 
70
- | Method | PCK@0.1 | PCK@0.05 | Speed-Up | Memory (GB) | config | download |
71
- |-----------------------------------|--------:|---------:|---------:|------------:|----------------------------------------------------------------------|----------------------------------------------------------------|
72
- | Segformer (Original) | 95.16 | 87.61 | - | 13.5 | [config](hpe_configs/coco/B5/segformer-coco-b5-default.py) | [model](https://mediastore.rz.uni-augsburg.de/get/ZOgj2NmQLy/) |
73
- | Segformer++<sub>HQ</sub> (ours) | 94.97 | 87.35 | 0.97 | 13.1 | [config](hpe_configs/coco/B5/segformer-coco-b5-bsm-hq.py) | [model](https://mediastore.rz.uni-augsburg.de/get/oAH5IlPxG8/) |
74
- | Segformer++<sub>fast</sub> (ours) | 95.02 | 87.37 | 0.99 | 12.9 | [config](hpe_configs/coco/B5/segformer-coco-b5-bsm-fast.py) | [model](https://mediastore.rz.uni-augsburg.de/get/3E2mMNLAAn/) |
75
- | Segformer++<sub>2x2</sub> (ours) | 94.98 | 87.36 | 1.24 | 12.3 | [config](hpe_configs/coco/B5/segformer-coco-b5-n2d-2x2.py) | [model](https://mediastore.rz.uni-augsburg.de/get/rzlgKC5XLc/) |
76
 
77
  ## Usage
78
 
 
28
 
29
  The weights of the Segformer (Original) model were used to get the inference results.
30
 
31
+ | Method | mIoU | Speed-Up | download |
32
+ |-----------------------------------|------:|---------:|----------------------------------------------------------------|
33
+ | Segformer (Original) | 82.39 | - | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
34
+ | Segformer++<sub>HQ</sub> (ours) | 82.31 | 1.61 | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
35
+ | Segformer++<sub>fast</sub> (ours) | 82.04 | 1.94 | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
36
+ | Segformer++<sub>2x2</sub> (ours) | 81.96 | 1.90 | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
37
+ | Segformer (Downsampling) | 77.31 | 6.51 | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
38
 
39
  ### Training on Cityscapes (MiT-B5)
40
 
41
+ | Method | mIoU | Speed-Up | Memory (GB) | download |
42
+ |-----------------------------------|------:|---------:|-------------|-----------------------------------------------------------------|
43
+ | Segformer (Original) | 82.39 | - | 48.3 | [model](https://mediastore.rz.uni-augsburg.de/get/yzE65lzm6N/) |
44
+ | Segformer++<sub>HQ</sub> (ours) | 82.19 | 1.40 | 34.0 | [model](https://mediastore.rz.uni-augsburg.de/get/i8fY8uXJrV/ ) |
45
+ | Segformer++<sub>fast</sub> (ours) | 81.77 | 1.55 | 30.5 | [model](https://mediastore.rz.uni-augsburg.de/get/cmG974iAxt/ ) |
46
+ | Segformer++<sub>2x2</sub> (ours) | 82.38 | 1.63 | 31.1 | [model](https://mediastore.rz.uni-augsburg.de/get/p0uMKbw531/) |
47
+ | Segformer (Downsampling) | 79.24 | 2.95 | 10.0 | [model](https://mediastore.rz.uni-augsburg.de/get/73zkKSO21t/) |
48
 
49
  ### Training on ADE20K (640x640) (MiT-B5)
50
 
51
+ | Method | mIoU | Speed-Up | Memory (GB) | download |
52
+ |-----------------------------------|------:|---------:|------------:|----------------------------------------------------------------|
53
+ | Segformer (Original) | 49.72 | - | 33.7 | [model](https://mediastore.rz.uni-augsburg.de/get/nKEjUHNAfK/) |
54
+ | Segformer++<sub>HQ</sub> (ours) | 49.77 | 1.15 | 29.2 | [model](https://mediastore.rz.uni-augsburg.de/get/Odyie8usgj/) |
55
+ | Segformer++<sub>fast</sub> (ours) | 49.10 | 1.20 | 28.0 | [model](https://mediastore.rz.uni-augsburg.de/get/K0IGkx4O2s/) |
56
+ | Segformer++<sub>2x2</sub> (ours) | 49.35 | 1.26 | 27.2 | [model](https://mediastore.rz.uni-augsburg.de/get/w5_Pxx4Q5C/) |
57
+ | Segformer (Downsampling) | 46.71 | 1.89 | 12.4 | [model](https://mediastore.rz.uni-augsburg.de/get/dFVvZQL6iL/) |
58
 
59
  ### Training on JBD
60
 
61
+ | Method | PCK@0.1 | PCK@0.05 | Speed-Up | Memory (GB) | download |
62
+ |-----------------------------------|--------:|---------:|---------:|------------:|----------------------------------------------------------------|
63
+ | Segformer (Original) | 95.20 | 90.65 | - | 40.0 | [model](https://mediastore.rz.uni-augsburg.de/get/psolrWXLLp/) |
64
+ | Segformer++<sub>HQ</sub> (ours) | 95.18 | 90.51 | 1.19 | 36.0 | [model](https://mediastore.rz.uni-augsburg.de/get/jx1eyecMLF/) |
65
+ | Segformer++<sub>fast</sub> (ours) | 94.58 | 89.87 | 1.25 | 34.6 | [model](https://mediastore.rz.uni-augsburg.de/get/K0IGkx4O2s/) |
66
+ | Segformer++<sub>2x2</sub> (ours) | 95.17 | 90.16 | 1.27 | 33.4 | [model](https://mediastore.rz.uni-augsburg.de/get/HumKbSB1vI/) |
67
 
68
  ### Training on MS COCO
69
 
70
+ | Method | PCK@0.1 | PCK@0.05 | Speed-Up | Memory (GB) | download |
71
+ |-----------------------------------|--------:|---------:|---------:|------------:|----------------------------------------------------------------|
72
+ | Segformer (Original) | 95.16 | 87.61 | - | 13.5 | [model](https://mediastore.rz.uni-augsburg.de/get/ZOgj2NmQLy/) |
73
+ | Segformer++<sub>HQ</sub> (ours) | 94.97 | 87.35 | 0.97 | 13.1 | [model](https://mediastore.rz.uni-augsburg.de/get/oAH5IlPxG8/) |
74
+ | Segformer++<sub>fast</sub> (ours) | 95.02 | 87.37 | 0.99 | 12.9 | [model](https://mediastore.rz.uni-augsburg.de/get/3E2mMNLAAn/) |
75
+ | Segformer++<sub>2x2</sub> (ours) | 94.98 | 87.36 | 1.24 | 12.3 | [model](https://mediastore.rz.uni-augsburg.de/get/rzlgKC5XLc/) |
76
 
77
  ## Usage
78