From 6b084811e8dfffbdd1741e59e12b33142d445c32 Mon Sep 17 00:00:00 2001 From: natanielruiz <nruiz9@gatech.edu> Date: 星期一, 30 十月 2017 08:34:01 +0800 Subject: [PATCH] a word --- code/test_on_video_dlib.py | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/code/test_on_video_dlib.py b/code/test_on_video_dlib.py index 0fbb080..1c9ddf1 100644 --- a/code/test_on_video_dlib.py +++ b/code/test_on_video_dlib.py @@ -26,7 +26,7 @@ default=0, type=int) parser.add_argument('--snapshot', dest='snapshot', help='Path of model snapshot.', default='', type=str) - parser.add_argument('--facedetection_model', dest='facedetection_model', help='Path of DLIB face detection model.', + parser.add_argument('--face_model', dest='face_model', help='Path of DLIB face detection model.', default='', type=str) parser.add_argument('--video', dest='video_path', help='Path of video') parser.add_argument('--bboxes', dest='bboxes', help='Bounding box annotations of frames') @@ -57,7 +57,7 @@ model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 6, 3], 66) # Dlib face detection model - cnn_face_detector = dlib.cnn_face_detection_model_v1(args.facedetection_model) + cnn_face_detector = dlib.cnn_face_detection_model_v1(args.face_model) print 'Loading snapshot.' # Load snapshot -- Gitblit v1.8.0