def test_cut_and_classify():
model_name = "model1.pkl"
report_name = "report.html"
# test cut
res, data_home = _cut(VIDEO_PATH)
# test train
_train(data_home, model_name)
# test classify
classify_result = _classify(VIDEO_PATH, data_home)
classify_result_1 = _classify(VIDEO_PATH, data_home, model=model_name)
# --- draw ---
r = Reporter()
r.draw(
classify_result,
report_path=os.path.join(data_home, report_name),
cut_result=res,
)
r.draw(
classify_result_1,
report_path=data_home,
cut_result=res,
)
def test_boost():
video = VideoObject(VIDEO_PATH)
video.load_frames()
# test cut
res, data_home = _cut(video)
# test classify
classify_result = _classify(video, data_home)
# --- draw ---
r = Reporter()
r.draw(
classify_result,
report_path=os.path.join(data_home, "report.html"),
cut_result=res,
)
# test compressing
r = Reporter()
r.draw(
classify_result,
report_path=os.path.join(data_home, "report.html"),
cut_result=res,
compress_rate=0.1,
)
r = Reporter()
r.draw(
classify_result,
report_path=os.path.join(data_home, "report.html"),
cut_result=res,
target_size=(600, 800),
)
def test_boost():
video = VideoObject(VIDEO_PATH)
video.load_frames()
# test cut
res, data_home = _cut(video)
# test classify
classify_result = _classify(video, data_home)
# --- draw ---
r = Reporter()
r.draw(
classify_result,
report_path=os.path.join(data_home, "report.html"),
cut_result=res,
)