def process(movie_path, record_dir_format, class_mapping):
try:
records = []
for label in class_mapping.items():
record = Record()
record.dir_format = record_dir_format
record.label = label
record.threshold = config["threshold"]
record.skip_frame_interval = config["skip_frame_interval"]
record.prepare()
records.append(record)
model = keras.models.load_model(model_file_path, compile=False)
predictor = Predictor(model, config["image_size_px"])
movie = Movie(movie_path, config["skip_frame_interval"], records, predictor)
if movie.is_completed_clip():
return
movie.capture()
movie.write_period_to_file()
except Exception as e:
print("record ERROR: ", movie_path)
print(e)
print(traceback.format_exc())
