摘要
目标检测中的数据增强是比较复杂,每一次改变图像同时也要考虑boxes的信息,比起目标分类更加局限性,比如翻转,左右翻转一般影响不大,但上下翻转造成的影响就截然不同。下面操作坐标点全是xyxy形式
resize操作
先来看下对比,在图片大小改变的同时也要改变boxes的位置信息,第一张是原图
我截图是一样大小,可以看出框的位置依旧准确,原图是(480,364)变化后的是(300,300),对比之前的照片清晰度明显下降,所以训练照片差距太多再好的模型也难以起到很好的效果。下面是代码实现
import math
import random
import torch
from PIL imp
1