We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 47f6bf3 commit 1b0fdfeCopy full SHA for 1b0fdfe
1 file changed
utils/utils.py
@@ -78,8 +78,8 @@ def preprocess(*image_path, max_size=512, mean=(0.485, 0.456, 0.406), std=(0.229
78
79
def preprocess_video(*frame_from_video, max_size=512, mean=(0.406, 0.456, 0.485), std=(0.225, 0.224, 0.229)):
80
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,
+ normalized_imgs = [(img[..., ::-1] / 255 - mean) / std for img in ori_imgs]
+ imgs_meta = [aspectaware_resize_padding(img, max_size, max_size,
83
means=None) for img in normalized_imgs]
84
framed_imgs = [img_meta[0] for img_meta in imgs_meta]
85
framed_metas = [img_meta[1:] for img_meta in imgs_meta]
0 commit comments