Update cloud38.py

icey-zhang · Jun 4, 2024 · 592ba20 · 592ba20
1 parent 03f40dc
commit 592ba20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/datasets/cloud38.py b/lib/datasets/cloud38.py
@@ -13,7 +13,7 @@
 from .base_dataset import BaseDataset
 import scipy.io as sio
 
-class Cityscapes(BaseDataset):
+class Cloud38(BaseDataset):
     def __init__(self, 
                  root, 
                  list_path, 
@@ -29,7 +29,7 @@ class Cityscapes(BaseDataset):
                  mean=[0.485, 0.456, 0.406], 
                  std=[0.229, 0.224, 0.225]):
 
-        super(Cityscapes, self).__init__(ignore_label, base_size,
+        super(Cloud38, self).__init__(ignore_label, base_size,
                 crop_size, downsample_rate, scale_factor, mean, std,)
 
         self.root = root