求解!为什么使用yolo训练神经网络的时候图片会自动旋转?

这几天训练神经网络的是否发现一个很离谱的事情,我的图片在被网络读取后会自动旋转(清除过exif信息)

使用的代码是github上的ultralytics/yolov5.6.1代码(经典)
用train文件训练时,有些图片在清除exif信息后被读取依旧会自动旋转
比如这张是本来的图片

img


这张是被batch后的图片

img


真的很迷惑,因为我要识别的是数字,要避免旋转,因为2旋转后是5,5旋转后是2。导致训练的时候出大问题

data/hyps/hpy.scartch.xxxx.yaml文件中有个fliplr这个超参数,不需要翻转的话改成0。如果改完之后还翻转,说明你改了增强了,utils/augmentations.py这个文件Albumentations这个类中的T列表改回原来的参数

  • 关于该问题,我找了一篇非常好的博客,你可以看看是否有帮助,链接:YOLO神经网络