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])