Clipping prior boxes in point form

Author: d-li14

layers/functions/prior_box.py

@@ -2,6 +2,7 @@
 from math import sqrt as sqrt
 from itertools import product as product
 import torch
+from layers.box_utils import point_form, center_size
 
 
 class PriorBox(object):
@@ -51,5 +52,7 @@ def forward(self):
         # back to torch land
         output = torch.Tensor(mean).view(-1, 4)
         if self.clip:
+            output = point_form(output)
             output.clamp_(max=1, min=0)
+            output = center_size(output)
         return output