def still_image_YOLO_DET(frames_list, frames_name, folder_path_det_frames,
folder_path_det_result):
print("Starting DET Phase")
if not os.path.exists(folder_path_det_frames):
os.makedirs(folder_path_det_frames)
print("Created Folder: %s" % folder_path_det_frames)
if not os.path.exists(folder_path_det_result):
os.makedirs(folder_path_det_result)
print("Created Folder: %s" % folder_path_det_result)
yolo = YOLO_small_tf.YOLO_TF()
det_frames_list = []
det_result_list = []
print("%d Frames to DET" % len(frames_list))
progress = progressbar.ProgressBar(widgets=[
progressbar.Bar('=', '[', ']'), ' ',
progressbar.Percentage(), ' ',
progressbar.ETA()
])
timer = Timer()
for i in progress(range(0, len(frames_list))):
# det_frame_name = frames_name[i]
#print frames_name[i]
timer.tic()
det_frame_name = frames_name[i].replace('.jpg', '_det.jpg')
det_frame_name = folder_path_det_frames + det_frame_name
det_frames_list.append(det_frame_name)
det_result_name = frames_name[i].replace('.jpg', '.txt')
det_result_name = folder_path_det_result + det_result_name
det_result_list.append(det_result_name)
yolo.tofile_txt = det_result_name
yolo.filewrite_txt = True
yolo.disp_console = False
yolo.filewrite_img = True
yolo.tofile_img = det_frame_name
yolo.detect_from_cvmat(frames_list[i][1])
timer.toc()
print('Detection took {:.3f}s ').format(timer.total_time)
return det_frames_list, det_result_list
import sys
import YOLO_small_tf
yolo = YOLO_small_tf.YOLO_TF()
yolo.disp_console = True
if len(sys.argv) > 2:
yolo.tofile_img = sys.argv[1] + '.processed.jpg'
yolo.filewrite_img = True
yolo.filewrite_txt = True
yolo.tofile_txt = sys.argv[1] + '.processed.txt'
yolo.detect_from_file(sys.argv[1])
yolo.tofile_img = sys.argv[2] + '.processed.jpg'
yolo.tofile_txt = sys.argv[2] + '.processed.txt'
yolo.detect_from_file(sys.argv[2])