Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Window mode and Video suppport(Separate Files) #7

Merged
merged 12 commits into from
Apr 7, 2018
Prev Previous commit
Merge branch 'master' into master
  • Loading branch information
qqwweee authored Apr 7, 2018
commit e7ebe6247c0f1858fea559dc9928ee6d0e7dca9b
2 changes: 1 addition & 1 deletion yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,6 +132,7 @@ def detect_image(self, image):
def close_session(self):
self.sess.close()


def detect_video(yolo,video_path):
vid = cv2.VideoCapture(video_path) ### TODO: will video path other than 0 be used?
if not vid.isOpened():
Expand All @@ -144,7 +145,6 @@ def detect_video(yolo,video_path):
return_value, frame = vid.read()
image = Image.fromarray(frame)
image = yolo.detect_image(image)

result = np.asarray(image)
curr_time = timer()
exec_time = curr_time - prev_time
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.