Update demo.py

facebookresearch · Aug 5, 2022 · 98502a6 · 98502a6
1 parent 964868d
commit 98502a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/demo.py b/demo/demo.py
@@ -67,7 +67,7 @@ def parse_args():
 # prepare input image
 transform = transforms.ToTensor()
 img_path = 'input.jpg'
-original_img = cv2.imread(img_path)
+original_img = load_img(img_path)
 original_img_height, original_img_width = original_img.shape[:2]
 
 # prepare bbox