只是修改了一下num_classes,模型并未改变。 尝试了一下,发现确实是因为输入图像的shape问题,必须得要w和h相等且等于模型定义时候的shape才不会报错。 我们跑的数据是64*1024的,可以的话,能否方便告知如果要修改的话大概需要修改哪里?如何修改? 期待在未来版本中针对输入图像的size问题进行优化!展开
只是修改了一下num_classes,模型并未改变。 尝试了一下,发现确实是因为输入图像的shape问题,必须得要w和h相等且等于模型定义时候的shape才不会报错。 我们跑的数据是64*1024的,可以的话,能否方便告知如果要修改的话大概需要修改哪里?如何修改? 期待在未来版本中针对输入图像的size问题进行优化!展开
我们本地复现了一下,模型是torchvision预定义加载的,输入的w和h要求一样,不一样就会报错,不知您这份模型是不是自定义修改过,如果官方定义的模型,请确认模型是否和你的输入shape匹配展开
请登录后评论