fix random image id random generation

src/utils.py

@@ -10,7 +10,13 @@ config = yaml.safe_load(open("./config/config.yaml"))
 def get_random_image_id(class_idx, data_dir):
     path = os.path.join(data_dir, 'images', str(class_idx))
     images = os.listdir(path)
-    return np.random.randint(0, len(images))
+    ids = [int(img.split('.')[0]) for img in images if img.endswith('.png')]
+    if not ids:
+        raise ValueError(f"No images found for class index {class_idx} in {path}")
+    # set random seed using time
+    np.random.seed(int(time.time()))
+    random_id = np.random.randint(0, len(ids))
+    return ids[random_id]
 
 def load_image_and_saliency(class_idx, data_dir, img_id):
     path = os.path.join(data_dir, 'images', str(class_idx))