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