Edit a comment

Input data shape is [batch,3,300,300] and not [batch,3*batch,300,300].
This small mistake makes some confusion.

ssd.py

@@ -50,7 +50,7 @@ def forward(self, x):
         """Applies network layers and ops on input image(s) x.
 
         Args:
-            x: input image or batch of images. Shape: [batch,3*batch,300,300].
+            x: input image or batch of images. Shape: [batch,3,300,300].
 
         Return:
             Depending on phase: