From 02c5c69ce019df7b5a027e2dfb02be29b71dabfb Mon Sep 17 00:00:00 2001 From: ekka Date: Wed, 13 Feb 2019 20:57:59 +0530 Subject: [PATCH 1/2] Updated inceptionV3 to accept different sized images (Adaptive avg pool) The update allows inceptionV3 to process images larger or smaller than prescribed image size (299x299) using adaptive average pooling. Will be useful while finetuning or testing on different resolution images. --- torchvision/models/inception.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torchvision/models/inception.py b/torchvision/models/inception.py index 7e7f27bbde0..5becf953560 100644 --- a/torchvision/models/inception.py +++ b/torchvision/models/inception.py @@ -118,7 +118,7 @@ def forward(self, x): # 8 x 8 x 2048 x = self.Mixed_7c(x) # 8 x 8 x 2048 - x = F.avg_pool2d(x, kernel_size=8) + x = F.adaptive_avg_pool2d(x, (1, 1)) # 1 x 1 x 2048 x = F.dropout(x, training=self.training) # 1 x 1 x 2048 @@ -310,6 +310,8 @@ def forward(self, x): x = self.conv0(x) # 5 x 5 x 128 x = self.conv1(x) + # Adaptive average pooling + x = F.adaptive_avg_pool2d(x, (1, 1)) # 1 x 1 x 768 x = x.view(x.size(0), -1) # 768 From 42379eb916c6ed1f79a0000da026612cdaf0bd08 Mon Sep 17 00:00:00 2001 From: ekka Date: Wed, 13 Feb 2019 21:10:14 +0530 Subject: [PATCH 2/2] Update inception.py --- torchvision/models/inception.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchvision/models/inception.py b/torchvision/models/inception.py index 5becf953560..9c49944a73a 100644 --- a/torchvision/models/inception.py +++ b/torchvision/models/inception.py @@ -118,6 +118,7 @@ def forward(self, x): # 8 x 8 x 2048 x = self.Mixed_7c(x) # 8 x 8 x 2048 + # Adaptive average pooling x = F.adaptive_avg_pool2d(x, (1, 1)) # 1 x 1 x 2048 x = F.dropout(x, training=self.training) @@ -310,6 +311,7 @@ def forward(self, x): x = self.conv0(x) # 5 x 5 x 128 x = self.conv1(x) + # 1 x 1 x 768 # Adaptive average pooling x = F.adaptive_avg_pool2d(x, (1, 1)) # 1 x 1 x 768