natanielruiz
2017-10-30 af57311e2e7f8b075b4aaa6d3e5fce3f777ebdfd
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