Update utils.py (#1)

- Update utils.py (8b4b482bf05346879e1f42575cfa68eba4d6d9bd)


Co-authored-by: Hongyang Wei <hongywei@users.noreply.huggingface.co>

utils.py

@@ -198,7 +198,7 @@ def postprocess(cfg, img, outputs, scale, resize, confidence_threshold, nms_thre
     Returns: 
         detetcion results
     """
-    _, _, im_height, im_width= img.shape
+    _,  im_height, im_width, _= img.shape
     loc = torch.from_numpy(outputs[0])
     conf = torch.from_numpy(outputs[1])
     landms = torch.from_numpy(outputs[2])
@@ -215,8 +215,8 @@ def postprocess(cfg, img, outputs, scale, resize, confidence_threshold, nms_thre
     scores = conf.squeeze(0).data.cpu().numpy()[:, 1]
     landms = decode_landm(landms.squeeze(0), prior_data, cfg["variance"])
     scale1 = torch.Tensor(
-        [img.shape[3], img.shape[2], img.shape[3], img.shape[2], img.shape[3],
-         img.shape[2], img.shape[3], img.shape[2], img.shape[3], img.shape[2],]
+        [img.shape[2], img.shape[1], img.shape[2], img.shape[1], img.shape[2],
+         img.shape[1], img.shape[2], img.shape[1], img.shape[2], img.shape[1],]
     )
     scale1 = scale1.to(device)
     landms = landms * scale1 / resize