最近在做图像分类时查阅了一些相关资料发现输入的图片大小并非是固定大小224*224,因此很疑惑,查阅了一些资料发现了以下解答:

版本框架:pytorch,模型加载来自torchvision

 对应链接为:https://discuss.pytorch.org/t/how-can-torchvison-models-deal-with-image-whose-size-is-not-224-224/51077

进一步的pytorch是如何做到自适应输入图片大小的?答案在AdaptiveAvgPool2d层,如下:

知乎的链接写的很全,可以详细看下:https://www.zhihu.com/question/282046628

希望对疑惑的小伙伴有所帮助

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐