最近在做图像分类时查阅了一些相关资料发现输入的图片大小并非是固定大小224*224,因此很疑惑,查阅了一些资料发现了以下解答:
版本框架:pytorch,模型加载来自torchvision
进一步的pytorch是如何做到自适应输入图片大小的?答案在AdaptiveAvgPool2d层,如下:
知乎的链接写的很全,可以详细看下:https://www.zhihu.com/question/282046628
希望对疑惑的小伙伴有所帮助
最近在做图像分类时查阅了一些相关资料发现输入的图片大小并非是固定大小224*224,因此很疑惑,查阅了一些资料发现了以下解答:
版本框架:pytorch,模型加载来自torchvision
进一步的pytorch是如何做到自适应输入图片大小的?答案在AdaptiveAvgPool2d层,如下:
知乎的链接写的很全,可以详细看下:https://www.zhihu.com/question/282046628
希望对疑惑的小伙伴有所帮助