Skip to content

Commit

Permalink
update results
Browse files Browse the repository at this point in the history
  • Loading branch information
@lzyhha
lzyhha committed Jan 12, 2023
1 parent e88d86c commit 8c78c40
Show file tree
Hide file tree
Showing 16 changed files with 140 additions and 58 deletions.
18 changes: 13 additions & 5 deletions configs/rfnext/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,21 +15,29 @@ The source code is publicly available on [http://mmcheng.net/rfnext](http://mmch

| Backbone | Method | RFNext | Crop Size | Lr Schd | mIoU | mIoU (ms+flip) | config | download |
| :-----------: | :----------------: | :-------------: | :-----: | :-----: | :-----: | :------: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| ConvNeXt-T | UPerNet | NO | 512x512 | 160000 | 46.11 | 46.62 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k/upernet_convnext_tiny_fp16_512x512_160k_ade20k_20220227_124553-cad485de.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k/upernet_convnext_tiny_fp16_512x512_160k_ade20k_20220227_124553.log.json) |
| RF-ConvNext-T | UperNet | Single-Branch | 512x512 | 160000 | | | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-ConvNext-T | UperNet | Multiple-Branch | 512x512 | 160000 | 46.78 | 47.70 | [search]() \| [retrain]() | [model]() \| [log]() |
| ConvNeXt-B | UPerNet | NO | 640x640 | 160000 | 52.13 | 52.66 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k/upernet_convnext_tiny_fp16_512x512_160k_ade20k_20220227_124553-cad485de.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k/upernet_convnext_tiny_fp16_512x512_160k_ade20k_20220227_124553.log.json) |
| RF-ConvNext-B | UperNet | Single-Branch | 640x640 | 160000 | | | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-ConvNext-B | UperNet | Single-Branch | 640x640 | 160000 | 52.51 | 53.12 | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-ConvNext-B | UperNet | Multiple-Branch | 640x640 | 160000 | 52.49 | 53.00 | [search]() \| [retrain]() | [model]() \| [log]() |

### DeepLabV3 on ADE20K

| Backbone | Method | RFNext | Crop Size | Lr Schd | mIoU | mIoU (ms+flip) | config | download |
| :-----------: | :----------------: | :-------------: | :-----: | :-----: | :-----: | :------: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-101-D8 | DeepLabV3 | NO | 512x512 | 80000 | 44.08 | 45.19 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k/deeplabv3_r101-d8_512x512_80k_ade20k_20200615_021256-d89c7fa4.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k/deeplabv3_r101-d8_512x512_80k_ade20k_20200615_021256.log.json) |
| RF-R-101-D8 | DeepLabV3 | Single-Branch | 512x512 | 80000 | | | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-R-101-D8 | DeepLabV3 | Single-Branch | 512x512 | 80000 | 44.65 | 45.65 | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-R-101-D8 | DeepLabV3 | Multiple-Branch | 512x512 | 80000 | 44.77 | 45.86 | [search]() \| [retrain]() | [model]() \| [log]() |

### DeepLabV3 on Pascal VOC 2012 + Aug

| Backbone | Method | RFNext | Crop Size | Lr Schd | mIoU | mIoU (ms+flip) | config | download |
| :-----------: | :----------------: | :-------------: | :-----: | :-----: | :-----: | :------: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | :----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| R-50-D8 | DeepLabV3 | NO | 512x512 | 20000 | 76.17 | 77.42 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug/deeplabv3_r50-d8_512x512_20k_voc12aug_20200617_010906-596905ef.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug/deeplabv3_r50-d8_512x512_20k_voc12aug_20200617_010906.log.json) |
| RF-R-50-D8 | DeepLabV3 | Single-Branch | 512x512 | 20000 | 77.48 | 78.83 | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-R-50-D8 | DeepLabV3 | Multiple-Branch | 512x512 | 20000 | 77.60 | 78.67 | [search]() \| [retrain]() | [model]() \| [log]() |
| R-101-D8 | DeepLabV3 | NO | 512x512 | 20000 | 78.70 | 79.95 | [config](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k.py) | [model](https://download.openmmlab.com/mmsegmentation/v0.5/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k/deeplabv3_r101-d8_512x512_80k_ade20k_20200615_021256-d89c7fa4.pth) \| [log](https://download.openmmlab.com/mmsegmentation/v0.5/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k/deeplabv3_r101-d8_512x512_80k_ade20k_20200615_021256.log.json) |
| RF-R-101-D8 | DeepLabV3 | Single-Branch | 512x512 | 20000 | 79.51 | 80.82 | [search]() \| [retrain]() | [model]() \| [log]() |
| RF-R-101-D8 | DeepLabV3 | Multiple-Branch | 512x512 | 20000 | 79.55 | 80.68 | [search]() \| [retrain]() | [model]() \| [log]() |

**Note:** The performance of multi-branch models listed above are evaluated during searching to save computional cost, retraining would achieve similar or better performance.

## Configs
Expand Down
23 changes: 23 additions & 0 deletions configs/rfnext/rfnext_fixed_multi_branch_deeplabv3_r101-d8_512x512_20k_voc12aug.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
_base_ = '../deeplabv3/deeplabv3_r101-d8_512x512_20k_voc12aug.py'

custom_hooks = [
dict(
type='RFSearchHook',
mode='fixed_multi_branch',
rfstructure_file='./configs/rfnext/search_log/deeplabv3_r101-d8_512x512_20k_voc12aug/local_search_config_step16000.json',
verbose=True,
by_epoch=False,
config=dict(
search=dict(
step=0,
max_step=16001,
search_interval=2000,
exp_rate=0.15,
init_alphas=0.01,
mmin=1,
mmax=64,
num_branches=3,
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]

find_unused_parameters=True
2 changes: 1 addition & 1 deletion configs/rfnext/rfnext_fixed_multi_branch_deeplabv3_r101-d8_512x512_80k_ade20k.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
dict(
type='RFSearchHook',
mode='fixed_multi_branch',
rfstructure_file='./config/rfnext/deeplabv3_r101-d8_512x512_80k_ade20k/local_search_config_step64000.json',
rfstructure_file='./configs/rfnext/search_log/deeplabv3_r101-d8_512x512_80k_ade20k/local_search_config_step64000.json',
verbose=True,
by_epoch=False,
config=dict(
Expand Down
23 changes: 23 additions & 0 deletions configs/rfnext/rfnext_fixed_multi_branch_deeplabv3_r50-d8_512x512_20k_voc12aug.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
_base_ = '../deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug.py'

custom_hooks = [
dict(
type='RFSearchHook',
mode='fixed_multi_branch',
rfstructure_file='./configs/rfnext/search_log/deeplabv3_r50-d8_512x512_20k_voc12aug/local_search_config_step16000.json',
verbose=True,
by_epoch=False,
config=dict(
search=dict(
step=0,
max_step=16001,
search_interval=2000,
exp_rate=0.15,
init_alphas=0.01,
mmin=1,
mmax=64,
num_branches=3,
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]

find_unused_parameters=True
21 changes: 0 additions & 21 deletions configs/rfnext/rfnext_fixed_multi_branch_upernet_convnext_tiny_fp16_512x512_160k_ade20k.py
View file

This file was deleted.

21 changes: 21 additions & 0 deletions configs/rfnext/rfnext_fixed_single_branch_deeplabv3_r101-d8_512x512_20k_voc12aug.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
_base_ = '../deeplabv3/deeplabv3_r101-d8_512x512_20k_voc12aug.py'

custom_hooks = [
dict(
type='RFSearchHook',
mode='fixed_single_branch',
rfstructure_file='./configs/rfnext/search_log/deeplabv3_r101-d8_512x512_20k_voc12aug/local_search_config_step16000.json',
verbose=True,
by_epoch=False,
config=dict(
search=dict(
step=0,
max_step=16001,
search_interval=2000,
exp_rate=0.15,
init_alphas=0.01,
mmin=1,
mmax=64,
num_branches=3,
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]
6 changes: 2 additions & 4 deletions configs/rfnext/rfnext_fixed_single_branch_deeplabv3_r101-d8_512x512_80k_ade20k.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
dict(
type='RFSearchHook',
mode='fixed_single_branch',
rfstructure_file='./config/rfnext/deeplabv3_r101-d8_512x512_80k_ade20k/local_search_config_step64000.json',
rfstructure_file='./configs/rfnext/search_log/deeplabv3_r101-d8_512x512_80k_ade20k/local_search_config_step64000.json',
verbose=True,
by_epoch=False,
config=dict(
Expand All @@ -18,6 +18,4 @@
mmax=64,
num_branches=3,
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]

find_unused_parameters=True
]
21 changes: 21 additions & 0 deletions configs/rfnext/rfnext_fixed_single_branch_deeplabv3_r50-d8_512x512_20k_voc12aug.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
_base_ = '../deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug.py'

custom_hooks = [
dict(
type='RFSearchHook',
mode='fixed_single_branch',
rfstructure_file='./configs/rfnext/search_log/deeplabv3_r50-d8_512x512_20k_voc12aug/local_search_config_step16000.json',
verbose=True,
by_epoch=False,
config=dict(
search=dict(
step=0,
max_step=16001,
search_interval=2000,
exp_rate=0.15,
init_alphas=0.01,
mmin=1,
mmax=64,
num_branches=3,
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]
1 change: 1 addition & 0 deletions configs/rfnext/rfnext_fixed_single_branch_upernet_convnext_base_fp16_640x640_160k_ade20k.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -19,3 +19,4 @@
num_branches=3,
skip_layer=['stages.0', 'stages.1', 'stages.2', 'auxiliary_head'])))
]
cudnn_benchmark=False
21 changes: 0 additions & 21 deletions configs/rfnext/rfnext_fixed_single_branch_upernet_convnext_tiny_fp16_512x512_160k_ade20k.py
View file

This file was deleted.

23 changes: 23 additions & 0 deletions configs/rfnext/rfnext_search_deeplabv3_r101-d8_512x512_20k_voc12aug.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
_base_ = '../deeplabv3/deeplabv3_r101-d8_512x512_20k_voc12aug.py'

custom_hooks = [
dict(
type='RFSearchHook',
mode='search',
rfstructure_file=None,
verbose=True,
by_epoch=False,
config=dict(
search=dict(
step=0,
max_step=16001,
search_interval=2000,
exp_rate=0.15,
init_alphas=0.01,
mmin=1,
mmax=64,
num_branches=3,
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]

find_unused_parameters=True
14 changes: 8 additions & 6 deletions ...convnext_tiny_fp16_512x512_160k_ade20k.py → ..._deeplabv3_r50-d8_512x512_20k_voc12aug.py
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
_base_ = '../convnext/upernet_convnext_tiny_fp16_512x512_160k_ade20k.py'
_base_ = '../deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug.py'

custom_hooks = [
dict(
Expand All @@ -10,12 +10,14 @@
config=dict(
search=dict(
step=0,
max_step=64001,
search_interval=8000,
exp_rate=0.5,
max_step=16001,
search_interval=2000,
exp_rate=0.15,
init_alphas=0.01,
mmin=1,
mmax=24,
mmax=64,
num_branches=3,
skip_layer=['stages.0', 'stages.1', 'stages.2', 'auxiliary_head'])))
skip_layer=['stem', 'conv1', 'layer1', 'layer2', 'layer3', 'auxiliary_head'])))
]

find_unused_parameters=True
Loading

0 comments on commit 8c78c40

Please sign in to comment.