如何更改输入模型大小

Canaan
,
1

老师，您好
按照教程中来最后的模型输出应该是40，20，20的大小用于检测
image
但是目前，我对yolov5进行了更改，输出模型大小变为了80，40，20
image
我现在运行代码会出现模型大小不匹配的问题，请问这里我该如何修改代码，来正确运行我的新模型

2

老师，打错了，教程中模型输出大小应该是40，20，10

3

老师，您好，可以回答下嘛，这个问题目前还是没能解决