Update model_configs/CVRP_mask2former.py

model_configs/CVRP_mask2former.py

@@ -96,8 +96,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,
@@ -424,7 +424,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=(
@@ -434,7 +434,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'))
@@ -459,7 +459,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'),
@@ -483,7 +483,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'))
@@ -540,7 +540,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=(
@@ -550,7 +550,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'))
@@ -569,4 +569,4 @@ visualizer = dict(
     vis_backends=[
         dict(type='LocalVisBackend'),
     ])
-work_dir = './work_dirs/TzyDataset-Mask2Former-0721'
+work_dir = './work_dirs/CVRPDataset_mask2former'