# Faster R-CNN 训练配置（torchvision）
# 运行: python train_faster_rcnn.py --cfg faster_rcnn.yaml

data: 01data.yaml
num_classes: 2              # 背景 + gongjian
epochs: 30
batch_size: 2
lr: 0.005
momentum: 0.9
weight_decay: 0.0005
step_size: 12
gamma: 0.1
num_workers: 2
device: 0
imgsz: 640                  # 短边缩放到该值附近
print_freq: 20
eval_every: 1
pretrained: true
project: runs/faster_rcnn
name: gongjian_frcnn
exist_ok: true
resume: ""                  # 断点 .pth 路径，空则从头训练