diff --git a/models/common.py b/models/common.py index 1653770..a0ee085 100644 --- a/models/common.py +++ b/models/common.py @@ -541,7 +541,7 @@ class DetectMultiBackend(nn.Module): elif self.coreml: # CoreML im = im.cpu().numpy() im = Image.fromarray((im[0] * 255).astype('uint8')) - # im = im.resize((192, 320), Image.ANTIALIAS) + # im = im.resize((192, 320), Image.BILINEAR) y = self.model.predict({'image': im}) # coordinates are xywh normalized if 'confidence' in y: box = xywh2xyxy(y['coordinates'] * [[w, h, w, h]]) # xyxy pixels