format
This commit is contained in:
23
configs/pennfudan_maskrcnn_config.py
Normal file
23
configs/pennfudan_maskrcnn_config.py
Normal 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)
|
||||
Reference in New Issue
Block a user