Update model_configs/CVRP_deeplabv3plus.py

model_configs/CVRP_deeplabv3plus.py

@@ -21,8 +21,8 @@ data_preprocessor = dict(
         57.375,
     ],
     type='SegDataPreProcessor')
-data_root = 'PanicleDataset/'
-dataset_type = 'TzyDataset'
+data_root = 'CVRPDataset/'
+dataset_type = 'CVRPDataset'
 default_hooks = dict(
     checkpoint=dict(
         by_epoch=False,
@@ -155,7 +155,7 @@ test_dataloader = dict(
     batch_size=1,
     dataset=dict(
         data_prefix=dict(img_path='img_dir/val', seg_map_path='ann_dir/val'),
-        data_root='PanicleDataset/',
+        data_root='CVRPDataset/',
         pipeline=[
             dict(type='LoadImageFromFile'),
             dict(keep_ratio=True, scale=(
@@ -165,7 +165,7 @@ test_dataloader = dict(
             dict(type='LoadAnnotations'),
             dict(type='PackSegInputs'),
         ],
-        type='TzyDataset'),
+        type='CVRPDataset'),
     num_workers=4,
     persistent_workers=True,
     sampler=dict(shuffle=False, type='DefaultSampler'))
@@ -190,7 +190,7 @@ train_dataloader = dict(
     dataset=dict(
         data_prefix=dict(
             img_path='img_dir/train', seg_map_path='ann_dir/train'),
-        data_root='PanicleDataset/',
+        data_root='CVRPDataset/',
         pipeline=[
             dict(type='LoadImageFromFile'),
             dict(type='LoadAnnotations'),
@@ -214,7 +214,7 @@ train_dataloader = dict(
             dict(type='PhotoMetricDistortion'),
             dict(type='PackSegInputs'),
         ],
-        type='TzyDataset'),
+        type='CVRPDataset'),
     num_workers=2,
     persistent_workers=True,
     sampler=dict(shuffle=True, type='InfiniteSampler'))
@@ -271,7 +271,7 @@ val_dataloader = dict(
     batch_size=1,
     dataset=dict(
         data_prefix=dict(img_path='img_dir/val', seg_map_path='ann_dir/val'),
-        data_root='PanicleDataset/',
+        data_root='CVRPDataset/',
         pipeline=[
             dict(type='LoadImageFromFile'),
             dict(keep_ratio=True, scale=(
@@ -281,7 +281,7 @@ val_dataloader = dict(
             dict(type='LoadAnnotations'),
             dict(type='PackSegInputs'),
         ],
-        type='TzyDataset'),
+        type='CVRPDataset'),
     num_workers=4,
     persistent_workers=True,
     sampler=dict(shuffle=False, type='DefaultSampler'))
@@ -300,4 +300,4 @@ visualizer = dict(
     vis_backends=[
         dict(type='LocalVisBackend'),
     ])
-work_dir = './work_dirs/TzyDataset-DeepLabV3plus-0725'
+work_dir = './work_dirs/CVRPDataset_DeepLabV3plus'