Update detection_utils.py
Browse files- detection_utils.py +1 -1
detection_utils.py
CHANGED
@@ -22,7 +22,7 @@ def predict_md(im,
|
|
22 |
# resize image
|
23 |
g = (size / max(im.size)) # multipl factor to make max size of the image equal to input size
|
24 |
im = im.resize((int(x * g) for x in im.size),
|
25 |
-
Image.
|
26 |
# device
|
27 |
if torch.cuda.is_available():
|
28 |
md_device = torch.device('cuda')
|
|
|
22 |
# resize image
|
23 |
g = (size / max(im.size)) # multipl factor to make max size of the image equal to input size
|
24 |
im = im.resize((int(x * g) for x in im.size),
|
25 |
+
PIL.Image.Resampling.LANCZOS) # resize
|
26 |
# device
|
27 |
if torch.cuda.is_available():
|
28 |
md_device = torch.device('cuda')
|