Skip to content

Commit 1545aad

Browse files
Update detection_demo.py
1 parent 1f59430 commit 1545aad

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

detection_demo.py

-4
Original file line numberDiff line numberDiff line change
@@ -15,17 +15,13 @@
1515
import tensorflow as tf
1616
from yolov3.utils import detect_image, detect_realtime, detect_video, Load_Yolo_model, detect_video_realtime_mp
1717
from yolov3.configs import *
18-
import time
1918

2019
image_path = "./IMAGES/kite.jpg"
2120
video_path = "./IMAGES/test.mp4"
2221

23-
t1 = time.time()
2422
yolo = Load_Yolo_model()
2523
detect_image(yolo, image_path, "./IMAGES/kite_pred.jpg", input_size=YOLO_INPUT_SIZE, show=True, rectangle_colors=(255,0,0))
2624
#detect_video(yolo, video_path, "", input_size=YOLO_INPUT_SIZE, show=False, rectangle_colors=(255,0,0))
2725
#detect_realtime(yolo, '', input_size=YOLO_INPUT_SIZE, show=True, rectangle_colors=(255, 0, 0))
2826

2927
#detect_video_realtime_mp(video_path, "Output.mp4", input_size=YOLO_INPUT_SIZE, show=False, rectangle_colors=(255,0,0), realtime=False)
30-
t2 = time.time()
31-
print(t2-t1)

0 commit comments

Comments
 (0)