在使用YOLOV8训练时,epoch训练到最后10次出现”Closing dataloader mosaic",又不是报错,但又不往下进行训练,有点懵了,后面经过了解,Yolov8是默认设置close_mosaic=10,需要把它修改为0;
close_mosaic: 禁用mosaic增强的最后第几个轮次。可以指定一个整数值,表示在训练的最后第几个轮次中禁用mosaic增强。;
原训练指令:
yolo task=segment mode=train model=yolov8s-seg.pt epochs=5 batch=1 data=co2_dataset.yaml
修改指令:
yolo task=segment mode=train model=yolov8s-seg.pt epochs=5 batch=1 close_mosaic=0 data=co2_dataset.yaml
同时大家也可以了解一下Mosaic(马赛克)数据增强算法的知识
Mosaic数据增强方法是YOLOV4论文中提出来的,主要思想是将四张图片进行随机裁剪,再拼接到一张图上作为训练数据。这样做的好处是丰富了图片的背景,并且四张图片拼接在一起变相地提高了batch_size,在进行batch normalization的时候也会计算四张图片,所以对本身batch_size不是很依赖,单块GPU就可以训练YOLOV4。
YOLOV4论文YOLOv4: Optimal Speed and Accuracy of Object Detection