Implement predict video
This commit is contained in:
parent
54c5987ef0
commit
4f5f79d32b
3
run.py
3
run.py
@ -88,12 +88,9 @@ def pre_check():
|
||||
|
||||
def start_processing():
|
||||
start_time = time.time()
|
||||
try:
|
||||
seconds, probabilities = predict_video_frames(video_path=args['target_path'], frame_interval=50)
|
||||
if any(probability > 0.7 for probability in probabilities):
|
||||
quit()
|
||||
except Exception as exception:
|
||||
quit(exception)
|
||||
if args['gpu']:
|
||||
process_video(args['source_img'], args["frame_paths"])
|
||||
end_time = time.time()
|
||||
|
Loading…
Reference in New Issue
Block a user