我正在尝试胶子模型动物园。
将mxnet导入为mx来自mxnet.gluon.model_zoo的导入愿景导入cv2导入numpy为np
ctx = mx.gpu(6)#successnet = vision.alexnet(pretrained = True,ctx = ctx)
#…
根据原始论文,你需要修改256到256的输入大小,因为这是AlexNet网络训练的图像大小。通常,您可以通过将较小的轴(宽度或高度)调整为256然后进行中心裁剪来实现。
问题在于,当您使用神经网络预测某些内容时,您需要以与训练数据完全相同的方式准备输入数据。如果不这样做,最简单的情况是会出现形状不匹配错误。在一个更复杂的情况下,当形状匹配,但图像与训练模型的图像大不相同时,结果肯定是错误的。
当我将输入图像调整到下面时 254*254 推论成功了。
254*254
也许 mxnet “训练前 alexnet 不处理大尺寸的图像。
mxnet
alexnet
@Sergei的评论有帮助。谢谢。