我们可以更改 YoloV7 中图像的输入尺寸吗？

Question

我正在尝试在 yolov7 中进行推理并更改其默认尺寸640x640至300x300.尽管模型权重是根据其默认维度进行训练的。我们可以改变它吗，我正在改变但它抛出张量错误？

path = 'image.png'
image = cv2.imread(path)
shape = image.shape
plt.imshow(image)
plt.show()
shape = image.shape
image = letterbox(image, 300, stride=64, auto=True)[0]
image_ = image.copy()

Answer 1

您应该使用在训练过程中使用的相同图像尺寸。如果您想使用 300x300 大小的图像，则应首先使用此尺寸训练模型。该模型仅接受在训练过程中使用的图像尺寸。如果您有不同的图像大小进行推理......在运行推理之前，应将它们调整为模型的输入数据大小。