Final test

prismer/experts/depth/generate_dataset.py

@@ -17,9 +17,7 @@ class Dataset(Dataset):
     def __init__(self, config, transform):
         self.data_path = config['data_path']
         self.transform = transform
-        # self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
-        data_folders = glob.glob(f'{self.data_path}/*/')
-        self.data_list = [data for f in data_folders for data in glob.glob(f + '*.jpg')]
+        self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
 
     def __len__(self):
         return len(self.data_list)

requirements.txt

@@ -9,7 +9,7 @@ pillow~=9.2.0
 torchvision~=0.15.1
 opencv-python-headless==4.7.0.72
 timm~=0.6.12
-matplotlib~=3.4.1
+matplotlib~=3.7.1
 pycocotools~=2.0.4
 scikit-learn==0.24.2
 yacs~=0.1.8