Refactor: Update Penn-Fudan Mask R-CNN configuration and data transformation logic for memory optimization

configs/pennfudan_maskrcnn_config.py

@@ -2,20 +2,30 @@
 Configuration for training Mask R-CNN on the Penn-Fudan dataset.
 """
 
-from .base_config import base_config
+from configs.base_config import base_config
 
+# Create a copy of the base configuration
 config = base_config.copy()
 
-# Override necessary settings from base_config
+# Update specific values for this experiment
 config.update(
     {
-        "config_name": "pennfudan_maskrcnn_v1",  # Unique name for this experiment run
-        "data_root": "data/PennFudanPed",  # Explicitly set dataset root
-        "num_classes": 2,  # Penn-Fudan has 1 class (pedestrian) + background
-        # Adjust other parameters as needed for this specific experiment, e.g.:
-        # 'batch_size': 4,
-        # 'num_epochs': 15,
-        # 'lr': 0.001,
+        # Core configuration
+        "config_name": "pennfudan_maskrcnn_v1",
+        "data_root": "data/PennFudanPed",
+        "num_classes": 2,  # background + pedestrian
+        # Training parameters - modified for memory constraints
+        "batch_size": 1,  # Reduced from 2 to 1 to save memory
+        "num_epochs": 10,
+        # Optimizer settings
+        "lr": 0.002,  # Slightly reduced learning rate for smaller batch size
+        "momentum": 0.9,
+        "weight_decay": 0.0005,
+        # Memory optimization settings
+        "pin_memory": False,  # Set to False to reduce memory pressure
+        "num_workers": 2,  # Reduced from 4 to 2
+        # Device settings
+        "device": "cuda",
     }
 )