update notebook

SainsburyWellcomeCentre · Jun 27, 2024 · a1c0a7c · a1c0a7c
1 parent e5616b8
commit a1c0a7c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/notebooks/notebook_data_augm.py b/notebooks/notebook_data_augm.py
@@ -6,9 +6,9 @@
 
 # %%%%%%%%%%%%%%%%%%%
 # Input data
-IMG_DIR = "/Users/sofia/arc/project_Zoo_crabs/sep2023-full/frames"
-ANNOT_FILE = "/Users/sofia/arc/project_Zoo_crabs/sep2023-full/annotations/VIA_JSON_combined_coco_gen.json"
-CONFIG = "/Users/sofia/arc/project_Zoo_crabs/restructure/crabs-exploration/crabs/detection_tracking/config/faster_rcnn.yaml"
+IMG_DIR = "/home/sminano/swc/project_crabs/data/sep2023-full/frames"
+ANNOT_FILE = "/home/sminano/swc/project_crabs/data/sep2023-full/annotations/VIA_JSON_combined_coco_gen.json"
+CONFIG = "/home/sminano/swc/project_crabs/crabs-exploration/crabs/detection_tracking/config/faster_rcnn.yaml"
 SPLIT_SEED = 42
 
 # %%%%%%%%%%%%%%%%%%%%
@@ -19,10 +19,10 @@
 # %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 # Create datamodule for the input data
 dm = CrabsDataModule(
-    [IMG_DIR],
-    [ANNOT_FILE],
-    config_dict,
-    SPLIT_SEED,
+    list_img_dirs=[IMG_DIR],
+    list_annotation_files=[ANNOT_FILE],
+    config=config_dict,
+    split_seed=SPLIT_SEED,
 )
 # %%%%%%%%%%%%%%%%%%%%%%%%
 # Setup for train / test