From af57311e2e7f8b075b4aaa6d3e5fce3f777ebdfd Mon Sep 17 00:00:00 2001 From: natanielruiz <nruiz9@gatech.edu> Date: 星期一, 30 十月 2017 08:35:08 +0800 Subject: [PATCH] a word --- code/test_on_video_dlib.py | 5 ++--- 1 files changed, 2 insertions(+), 3 deletions(-) diff --git a/code/test_on_video_dlib.py b/code/test_on_video_dlib.py index 0fbb080..d03415f 100644 --- a/code/test_on_video_dlib.py +++ b/code/test_on_video_dlib.py @@ -26,10 +26,9 @@ 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') parser.add_argument('--output_string', dest='output_string', help='String appended to output file') parser.add_argument('--n_frames', dest='n_frames', help='Number of frames', type=int) parser.add_argument('--fps', dest='fps', help='Frames per second of source video', type=float, default=30.) @@ -57,7 +56,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