From 55f6d16a5646ef5f5dc1b079f61e8f3a44c69256 Mon Sep 17 00:00:00 2001
From: natanielruiz <nruiz9@gatech.edu>
Date: 星期一, 30 十月 2017 08:36:55 +0800
Subject: [PATCH] a word
---
code/test_on_video_dlib.py | 13 +++++++------
1 files changed, 7 insertions(+), 6 deletions(-)
diff --git a/code/test_on_video_dlib.py b/code/test_on_video_dlib.py
index 1c9ddf1..31d364b 100644
--- a/code/test_on_video_dlib.py
+++ b/code/test_on_video_dlib.py
@@ -29,7 +29,6 @@
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.)
@@ -99,6 +98,8 @@
# fourcc = cv2.cv.CV_FOURCC(*'MJPG')
# out = cv2.VideoWriter('output/video/output-%s.avi' % args.output_string, fourcc, 30.0, (width, height))
+ txt_out = open('output/video/output-%s.txt' % args.output_string, 'w')
+
frame_num = 1
while frame_num <= args.n_frames:
@@ -113,11 +114,11 @@
for idx, det in enumerate(dets):
# Get x_min, y_min, x_max, y_max, conf
- x_min = d.rect.left()
- y_min = d.rect.top()
- x_max = d.rect.right()
- y_max = d.rect.bottom()
- conf = d.confidence
+ x_min = det.rect.left()
+ y_min = det.rect.top()
+ 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:
--
Gitblit v1.8.0