forked from open-mmlab/mmsegmentation
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Datasets] Add Mapillary Vistas Datasets to MMSeg Core Package. (open…
…-mmlab#2576) ## [Datasets] Add Mapillary Vistas Datasets to MMSeg Core Package . ## Motivation Add Mapillary Vistas Datasets to core package. Old PR open-mmlab#2484 ## Modification - Add Mapillary Vistas Datasets to core package. - Delete `tools/datasets_convert/mapillary.py` , dataset does't need converting. - Add `schedule_240k.py` config. - Add configs files. ```none deeplabv3plus_r101-d8_4xb2-240k_mapillay_v1-512x1024.py deeplabv3plus_r101-d8_4xb2-240k_mapillay_v2-512x1024.py maskformer_swin-s_4xb2-240k_mapillary_v1-512x1024.py maskformer_swin-s_4xb2-240k_mapillary_v2-512x1024.py maskformer_r101-d8_4xb2-240k_mapillary_v1-512x1024.py maskformer_r101-d8_4xb2-240k_mapillary_v2-512x1024.py pspnet_r101-d8_4xb2-240k_mapillay_v1-512x1024.py pspnet_r101-d8_4xb2-240k_mapillay_v2-512x1024.py ``` - Synchronized changes to `projects/mapillary_datasets` --------- Co-authored-by: Miao Zheng <[email protected]> Co-authored-by: xiexinch <[email protected]>
- Loading branch information
3 people
authored
Mar 15, 2023
1 parent
447a398
commit 8c89ff3
Showing
30 changed files
with
1,089 additions
and
518 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,37 @@ | ||
| # dataset settings | ||
| _base_ = './mapillary_v1.py' | ||
| metainfo = dict( | ||
| classes=('Bird', 'Ground Animal', 'Curb', 'Fence', 'Guard Rail', 'Barrier', | ||
| 'Wall', 'Bike Lane', 'Crosswalk - Plain', 'Curb Cut', 'Parking', | ||
| 'Pedestrian Area', 'Rail Track', 'Road', 'Service Lane', | ||
| 'Sidewalk', 'Bridge', 'Building', 'Tunnel', 'Person', 'Bicyclist', | ||
| 'Motorcyclist', 'Other Rider', 'Lane Marking - Crosswalk', | ||
| 'Lane Marking - General', 'Mountain', 'Sand', 'Sky', 'Snow', | ||
| 'Terrain', 'Vegetation', 'Water', 'Banner', 'Bench', 'Bike Rack', | ||
| 'Billboard', 'Catch Basin', 'CCTV Camera', 'Fire Hydrant', | ||
| 'Junction Box', 'Mailbox', 'Manhole', 'Phone Booth', 'Pothole', | ||
| 'Street Light', 'Pole', 'Traffic Sign Frame', 'Utility Pole', | ||
| 'Traffic Light', 'Traffic Sign (Back)', 'Traffic Sign (Front)', | ||
| 'Trash Can', 'Bicycle', 'Boat', 'Bus', 'Car', 'Caravan', | ||
| 'Motorcycle', 'On Rails', 'Other Vehicle', 'Trailer', 'Truck', | ||
| 'Wheeled Slow', 'Car Mount', 'Ego Vehicle'), | ||
| palette=[[165, 42, 42], [0, 192, 0], [196, 196, 196], [190, 153, 153], | ||
| [180, 165, 180], [90, 120, 150], [102, 102, 156], [128, 64, 255], | ||
| [140, 140, 200], [170, 170, 170], [250, 170, 160], [96, 96, 96], | ||
| [230, 150, 140], [128, 64, 128], [110, 110, 110], [244, 35, 232], | ||
| [150, 100, 100], [70, 70, 70], [150, 120, 90], [220, 20, 60], | ||
| [255, 0, 0], [255, 0, 100], [255, 0, 200], [200, 128, 128], | ||
| [255, 255, 255], [64, 170, 64], [230, 160, 50], [70, 130, 180], | ||
| [190, 255, 255], [152, 251, 152], [107, 142, 35], [0, 170, 30], | ||
| [255, 255, 128], [250, 0, 30], [100, 140, 180], [220, 220, 220], | ||
| [220, 128, 128], [222, 40, 40], [100, 170, 30], [40, 40, 40], | ||
| [33, 33, 33], [100, 128, 160], [142, 0, 0], [70, 100, 150], | ||
| [210, 170, 100], [153, 153, 153], [128, 128, 128], [0, 0, 80], | ||
| [250, 170, 30], [192, 192, 192], [220, 220, 0], [140, 140, 20], | ||
| [119, 11, 32], [150, 0, 255], [0, 60, 100], [0, 0, 142], | ||
| [0, 0, 90], [0, 0, 230], [0, 80, 100], [128, 64, 64], [0, 0, 110], | ||
| [0, 0, 70], [0, 0, 192], [32, 32, 32], [120, 10, 10]]) | ||
|
|
||
| train_dataloader = dict(dataset=dict(metainfo=metainfo)) | ||
| val_dataloader = dict(dataset=dict(metainfo=metainfo)) | ||
| test_dataloader = val_dataloader |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,25 @@ | ||
| # optimizer | ||
| optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) | ||
| optim_wrapper = dict(type='OptimWrapper', optimizer=optimizer, clip_grad=None) | ||
| # learning policy | ||
| param_scheduler = [ | ||
| dict( | ||
| type='PolyLR', | ||
| eta_min=1e-4, | ||
| power=0.9, | ||
| begin=0, | ||
| end=240000, | ||
| by_epoch=False) | ||
| ] | ||
| # training schedule for 240k | ||
| train_cfg = dict( | ||
| type='IterBasedTrainLoop', max_iters=240000, val_interval=24000) | ||
| val_cfg = dict(type='ValLoop') | ||
| test_cfg = dict(type='TestLoop') | ||
| default_hooks = dict( | ||
| timer=dict(type='IterTimerHook'), | ||
| logger=dict(type='LoggerHook', interval=50, log_metric_by_epoch=False), | ||
| param_scheduler=dict(type='ParamSchedulerHook'), | ||
| checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=24000), | ||
| sampler_seed=dict(type='DistSamplerSeedHook'), | ||
| visualization=dict(type='SegVisualizationHook')) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
58 changes: 58 additions & 0 deletions
58
configs/deeplabv3plus/deeplabv3plus_r50-d8_4xb2-300k_mapillay_v1_65-1280x1280.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,58 @@ | ||
| _base_ = [ | ||
| '../_base_/models/deeplabv3plus_r50-d8.py', | ||
| '../_base_/datasets/mapillary_v1_65.py', | ||
| '../_base_/default_runtime.py', | ||
| ] | ||
|
|
||
| crop_size = (1280, 1280) | ||
| data_preprocessor = dict(size=crop_size) | ||
| model = dict( | ||
| data_preprocessor=data_preprocessor, | ||
| pretrained='open-mmlab://resnet50_v1c', | ||
| backbone=dict(depth=50), | ||
| decode_head=dict(num_classes=65), | ||
| auxiliary_head=dict(num_classes=65)) | ||
|
|
||
| iters = 300000 | ||
| # optimizer | ||
| optimizer = dict( | ||
| type='AdamW', lr=0.0001, betas=(0.9, 0.999), weight_decay=0.0001) | ||
| # optimizer | ||
| optim_wrapper = dict( | ||
| type='OptimWrapper', | ||
| optimizer=optimizer, | ||
| clip_grad=dict(max_norm=0.01, norm_type=2), | ||
| paramwise_cfg=dict( | ||
| custom_keys={'backbone': dict(lr_mult=0.1, decay_mult=1.0)})) | ||
| param_scheduler = [ | ||
| dict( | ||
| type='PolyLR', | ||
| eta_min=0, | ||
| power=0.9, | ||
| begin=0, | ||
| end=iters, | ||
| by_epoch=False) | ||
| ] | ||
|
|
||
| # training schedule for 300k | ||
| train_cfg = dict( | ||
| type='IterBasedTrainLoop', max_iters=iters, val_interval=iters // 10) | ||
| val_cfg = dict(type='ValLoop') | ||
| test_cfg = dict(type='TestLoop') | ||
|
|
||
| default_hooks = dict( | ||
| timer=dict(type='IterTimerHook'), | ||
| logger=dict(type='LoggerHook', interval=50, log_metric_by_epoch=False), | ||
| param_scheduler=dict(type='ParamSchedulerHook'), | ||
| checkpoint=dict( | ||
| type='CheckpointHook', by_epoch=False, interval=iters // 10), | ||
| sampler_seed=dict(type='DistSamplerSeedHook'), | ||
| visualization=dict(type='SegVisualizationHook')) | ||
|
|
||
| train_dataloader = dict(batch_size=2) | ||
|
|
||
| # Default setting for scaling LR automatically | ||
| # - `enable` means enable scaling LR automatically | ||
| # or not by default. | ||
| # - `base_batch_size` = (4 GPUs) x (2 samples per GPU). | ||
| auto_scale_lr = dict(enable=False, base_batch_size=8) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.