Create urbansyn_synthia_gtav_to_citys.py

w1oves · Sep 27, 2024 · 9d09660 · 9d09660
1 parent 3f28097
commit 9d09660
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/configs/_base_/datasets/urbansyn_synthia_gtav_to_citys.py b/configs/_base_/datasets/urbansyn_synthia_gtav_to_citys.py
@@ -0,0 +1,31 @@
+_base_ = [
+    "./urbansyn_512x512.py",
+    "./synthia_512x512.py",
+    "./gta_512x512.py",
+    "./cityscapes_512x512.py",
+]
+train_dataloader = dict(
+    batch_size=2,
+    num_workers=2,
+    persistent_workers=True,
+    pin_memory=True,
+    sampler=dict(type="InfiniteSampler", shuffle=True),
+    dataset=dict(
+        type="ConcatDataset",
+        datasets=[
+            {{_base_.train_urbansyn}},
+            {{_base_.train_gta}},
+            {{_base_.train_syn}},
+        ],
+    ),
+)
+val_dataloader = dict(
+    batch_size=1,
+    num_workers=4,
+    persistent_workers=True,
+    sampler=dict(type="DefaultSampler", shuffle=False),
+    dataset={{_base_.val_cityscapes}},
+)
+test_dataloader = val_dataloader
+val_evaluator = dict(type="IoUMetric", iou_metrics=["mIoU"])
+test_evaluator = val_evaluator