diff --git a/Qwen_app.py b/Qwen_app.py index 97a68f4..a77cdf4 100644 --- a/Qwen_app.py +++ b/Qwen_app.py @@ -231,7 +231,7 @@ def run(vid_file: str, run_sam3: bool = True, run_inspection: bool = True, gen_r # 获取总帧数与帧率 cap = cv2.VideoCapture(input_video_path) total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) - fps = cap.get(cv2.CAP_PROP_FPS) + fps = int(cap.get(cv2.CAP_PROP_FPS)) cap.release() @@ -287,9 +287,9 @@ def run(vid_file: str, run_sam3: bool = True, run_inspection: bool = True, gen_r #================生成报告================= if gen_report: - if result == {}: - with open(f"{output_dir}/hazard_inspection.json", "r", encoding="utf-8") as f: - result = json.load(f) + result: dict = {} + with open(f"{output_dir}/hazard_inspection.json", "r", encoding="utf-8") as f: + result = json.load(f) with open(f"知识库/rule.json", "r", encoding="utf-8") as f: rule_definitions = json.load(f)