请问自己训练模型时，如何修改输入图片的尺寸？

JiayuanWang-JW / YOLOv8-multi-task

GNU Affero General Public License v3.0

234 stars 39 forks source link

请问自己训练模型时，如何修改输入图片的尺寸？ #77

Open wycrystal opened 1 month ago

wycrystal commented 1 month ago

您好！我将预训练模型v4.pt转成onnx模型后，用https://netron.app/ 查看模型的输入尺寸是 1x3x384x672, 如果我想重新训练时将这个尺寸改得更小一些，比如改成1x3x192x336，是不是可以使模型推理得更快一些？如果要改的话，都需要修改哪些地方，麻烦指导一下，非常感谢！