Yolov7使用

1. 安装和测试：

【小白教学】如何用YOLOv7训练自己的数据集 - 知乎

YOLOv7——训练自己的数据集 - 代码网

*** 这里需要注意如果 torch 和torchvion版本太高，会导致GPU不可用，因为GPU?drive版本太低。可以使用低版本的：?pip install torch==1.12.1 torchvision==0.13.1

将/home/xumin/yolov7/utils/dataloaders.py修改

#sa, sb = f'{os.sep}images{os.sep}', f'{os.sep}labels{os.sep}'  # /images/, /labels/ substrings
sa, sb = f'{os.sep}JPEGImages{os.sep}', f'{os.sep}labels{os.sep}'

其他和Yolov5一样（数据准备，配置等）。。。

需要注意的是

（1）yolov7.yaml中的参数设置要yolov5s一致（如果做对比实验）

# parameters
nc: 1 ?# number of classes
depth_multiple: 0.33 ?# model depth multiple
width_multiple: 0.50 ?# layer channel multiple

（2）?hyp.scratch-low.yaml可以直接copy和使用yolov5的

(3) 如果与yolov5对比，设置optimizer为SGD

Train commands:

python train.py --img 640 --batch 2 --epoch 600 --data data/aphid_voc.yaml --cfg cfg/training/yolov7.yaml --weights weights/yolov7.pt --save_period 100