YOLOv5进行水域游泳者检测系统, 编写数据集配置文件
修改位于 models/yolov5s_swimmer.yaml 的文件内容。
主要是修改 ==nc== 参数的值,这里我们有 ==person== 和 ==swimmer== 两种类型的目标,所以 nc 设置为 2。
# parameters
nc: 2 # number of classes
depth_multiple: 0.33 # model depth multiple
width_multiple: 0.50 # layer channel multiple
# anchors
anchors:
- [10,13, 16,30, 33,23] # P3/8
- [30,61, 62,45, 59,119] # P4/16
- [116,90, 156,198, 373,326] # P5/32
# YOLOv5 backbone
backbone:
# [from, number, module, args]
[[-1, 1, Focus, [64, 3]], # 0-P1