Nataniel Ruiz
2019-03-04 cd27a07f60e1896e93554f4e91152e61cf40b2b2
code/test_on_video_dlib.py
@@ -103,6 +103,8 @@
    frame_num = 1
    while frame_num <= args.n_frames:
        print frame_num
        ret,frame = video.read()
        if ret == False:
            break
@@ -119,13 +121,12 @@
            x_max = det.rect.right()
            y_max = det.rect.bottom()
            conf = det.confidence
            print x_min, y_min, x_max, y_max, conf
            if conf > 0.95:
            if conf > 1.0:
                bbox_width = abs(x_max - x_min)
                bbox_height = abs(y_max - y_min)
                x_min -= 3 * bbox_width / 4
                x_max += 3 * bbox_width / 4
                x_min -= 2 * bbox_width / 4
                x_max += 2 * bbox_width / 4
                y_min -= 3 * bbox_height / 4
                y_max += bbox_height / 4
                x_min = max(x_min, 0); y_min = max(y_min, 0)
@@ -157,8 +158,8 @@
                # Plot expanded bounding box
                # cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), (0,255,0), 1)
            out.write(frame)
            frame_num += 1
        out.write(frame)
        frame_num += 1
    out.release()
    video.release()