This commit is contained in:
Craig
2025-04-12 10:28:50 +01:00
parent ae79a555d3
commit c3096f0664
3 changed files with 53 additions and 5 deletions

View File

@@ -0,0 +1,23 @@
"""
Configuration for training Mask R-CNN on the Penn-Fudan dataset.
"""
from .base_config import base_config
config = base_config.copy()
# Override necessary settings from base_config
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,
}
)
# Ensure derived paths or settings are consistent if needed
# (Not strictly necessary with this simple structure)