笔记Yolov5 (4-Mosaic/9-Mosaic)

yolov5有关数据增强的参数都写到了data/hyps/hyp.scratch-low/med/high.yaml文件里,如果想关闭mosaic数据增强就直接可以把mosaic的参数设置为0

但是在Yolov5源码中有两种mosaic数据增强代码,一个是4-mosaic数据增强,另一个是9-mosaic数据增强,默认用的是4-mosaic数据增强,如果想换成9-mosaic数据增强可以将utils/datasets.py文件中的load_mosaic9()改成load_mosaic(),然后将原本的load_mosaic()注释掉,其他不变

reference

Yolo v5 (v6.1)数据增强方式解析_迪菲赫尔曼的博客-CSDN博客


版权声明:本文为m0_56247038原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。