diff --git a/pic_detect.py b/pic_detect.py index bcd788a..8fa6cb4 100644 --- a/pic_detect.py +++ b/pic_detect.py @@ -165,13 +165,15 @@ def pic_detect_func(mqtt_pub_ip, mqtt_pub_port, mqtt_pub_topic, try: frame_copy = frame.copy() results = counter(frame) + + func_id=model_func_id_list[0] annotated_frame, box_result = cal_tricker_results(frame_copy, counter, class_names, - model_func_id_list, + func_id, local_func_cache, para, cls, chinese_label, model_func_id_list[0]) except Exception as e: - print(f"处理帧错误: {e}") + print(f"处理帧错误1: {e}") error_count += 1 if error_count >= 5: print(f"连续处理错误达到5次 ,正在停止处理...") diff --git a/pt/build-wall.pt b/pt/build-wall.pt new file mode 100644 index 0000000..d494ba6 Binary files /dev/null and b/pt/build-wall.pt differ