PyTorch_YOLOv1
PyTorch_YOLOv1 copied to clipboard
Published
20 hours ago •
yjh0410
yjh0410
请问数据增强的时候，resize图片需要调整框的位置吗？
https://github.com/yjh0410/PyTorch_YOLOv1/blob/2335aee70380ea12606abe393f6782d28e0f84f9/utils/augmentations.py#L108 原文代码没有调整框的大小，经过数据增强后的图片，按照SSDAugmentation内的顺序去作数据增强，得到的图片框无法被准确画到目标上
@WuShaogui bbox的坐标已经被归一化了，所以不受resize的影响。
是的，我训练比较差异不大。那是不是意味着解析模型的框预测时，如果想绘制到原图就乘原图宽高，如果想绘制到数据增强后的图（即模型输入，经过resize）,就乘模型输入的宽高