修改模型尺寸

[zyz211314]

请问大佬，修改模型尺寸640改为320后为什么运行报错啊，除了shape[1,3,640,640]改为[1,3,320,320]还需要修改其他东西的。