Skip to content

Commit 7f236e0

Browse files
committedApr 19, 2021
removed bug
1 parent 0918d4e commit 7f236e0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed
 

‎yolo_inference.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import torch
77
from torchvision import transforms
88

9+
from emotic import Emotic
910
from inference import infer
1011
from yolo_utils import prepare_yolo, rescale_boxes, non_max_suppression
1112

@@ -98,7 +99,7 @@ def yolo_infer(images_list, result_path, model_path, context_norm, body_norm, in
9899

99100

100101
def yolo_video(video_file, result_path, model_path, context_norm, body_norm, ind2cat, ind2vad, args):
101-
''' Perform inference onn a video. First yolo model is used to obtain bounding boxes of persons in every frame.
102+
''' Perform inference on a video. First yolo model is used to obtain bounding boxes of persons in every frame.
102103
After that the emotic model is used to obtain categoraical and continuous emotion predictions.
103104
:param video_file: Path of video file.
104105
:param result_path: Directory path to save the results (output video).

0 commit comments

Comments
 (0)
Please sign in to comment.