Fix inception v3 input transform (#144)

First dimension is batch size, channel is the second

Author: lopuhin

torchvision/models/inception.py

@@ -70,9 +70,9 @@ def __init__(self, num_classes=1000, aux_logits=True, transform_input=False):
     def forward(self, x):
         if self.transform_input:
             x = x.clone()
-            x[0] = x[0] * (0.229 / 0.5) + (0.485 - 0.5) / 0.5
-            x[1] = x[1] * (0.224 / 0.5) + (0.456 - 0.5) / 0.5
-            x[2] = x[2] * (0.225 / 0.5) + (0.406 - 0.5) / 0.5
+            x[:, 0] = x[:, 0] * (0.229 / 0.5) + (0.485 - 0.5) / 0.5
+            x[:, 1] = x[:, 1] * (0.224 / 0.5) + (0.456 - 0.5) / 0.5
+            x[:, 2] = x[:, 2] * (0.225 / 0.5) + (0.406 - 0.5) / 0.5
         # 299 x 299 x 3
         x = self.Conv2d_1a_3x3(x)
         # 149 x 149 x 32