Skip to content

Commit 1b0fdfe

Browse files
committed
fix bgr order in display function
1 parent 47f6bf3 commit 1b0fdfe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,8 @@ def preprocess(*image_path, max_size=512, mean=(0.485, 0.456, 0.406), std=(0.229
7878

7979
def preprocess_video(*frame_from_video, max_size=512, mean=(0.406, 0.456, 0.485), std=(0.225, 0.224, 0.229)):
8080
ori_imgs = frame_from_video
81-
normalized_imgs = [(img / 255 - mean) / std for img in ori_imgs]
82-
imgs_meta = [aspectaware_resize_padding(img[..., ::-1], max_size, max_size,
81+
normalized_imgs = [(img[..., ::-1] / 255 - mean) / std for img in ori_imgs]
82+
imgs_meta = [aspectaware_resize_padding(img, max_size, max_size,
8383
means=None) for img in normalized_imgs]
8484
framed_imgs = [img_meta[0] for img_meta in imgs_meta]
8585
framed_metas = [img_meta[1:] for img_meta in imgs_meta]

0 commit comments

Comments
 (0)