first

2026-08-21 14:51:57 +08:00
commit 4ad53f4e97
2056 changed files with 6272 additions and 0 deletions
@@ -0,0 +1,22 @@
+# 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 路径，空则从头训练