orisys
23 lines
529 B
YAML
23 lines
529 B
YAML
# 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 路径，空则从头训练
|