智凌Brent 2022-11-28 08:14 采纳率: 100%
浏览 34
已结题

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

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

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

img


这张是被batch后的图片

img


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

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2022-11-28 09:11
    关注

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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 12月8日
  • 已采纳回答 11月30日
  • 创建了问题 11月28日