24 lines
728 B
Python
24 lines
728 B
Python
"""
|
|
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)
|