Merge pull request amdegroot#90 from jackiechen0708/master

add support for python2

Author: ellisbrown

layers/box_utils.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import torch
 
 def point_form(boxes):

layers/functions/prior_box.py

@@ -1,3 +1,4 @@
+from __future__ import division
 import torch
 from math import sqrt as sqrt
 from itertools import product as product

layers/modules/l2norm.py

@@ -18,6 +18,7 @@ def reset_parameters(self):
 
     def forward(self, x):
         norm = x.pow(2).sum(dim=1, keepdim=True).sqrt()+self.eps
-        x /= norm
+        #x /= norm
+        x = torch.div(x,norm)
         out = self.weight.unsqueeze(0).unsqueeze(2).unsqueeze(3).expand_as(x) * x
         return out

layers/modules/multibox_loss.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import torch
 import torch.nn as nn
 import torch.nn.functional as F

ssd.py

@@ -199,7 +199,7 @@ def build_ssd(phase, size=300, num_classes=21):
     if size != 300:
         print("Error: Sorry only SSD300 is supported currently!")
         return
-
-    return SSD(phase, *multibox(vgg(base[str(size)], 3),
-                                add_extras(extras[str(size)], 1024),
-                                mbox[str(size)], num_classes), num_classes)
+    base_,extras_,head_=multibox(vgg(base[str(size)], 3),
+                                 add_extras(extras[str(size)], 1024),
+                                 mbox[str(size)], num_classes)   
+    return SSD(phase,base_,extras_,head_, num_classes)