dc dc dc

src/utils.py

@@ -8,6 +8,7 @@ import time
 config = yaml.safe_load(open("./config/config.yaml")) 
     
 def get_random_image_id(class_idx, data_dir):
+    print(f'getting random image id for class index: {class_idx}')
     path = os.path.join(data_dir, 'images', str(class_idx))
     images = os.listdir(path)
     ids = [int(img.split('.')[0]) for img in images if img.endswith('.png')]