根据您提供的文档内容和错误提示 `[OCR异常] : image format not match, input_type: BGR888, image format: RGB888` ，可以看出问题是由于图像格式不匹配导致的。OCR 模块期望的是 `BGR888` 格式的图像，而当前提供的是 `RGB888` 格式。 为了修正这个问题，我们可以在调用 `ocr_processing` 函数之前将图像从 `RGB888` 转换为 `BGR888` 格式。以下是修改后的完整代码： ```python from maix import camera, display, image, nn, app import time, os # 初始化模型 detector = nn.YOLOv5(model="/root/models/mymodel/model_127448.mud", dual_buff=True) ocr = nn.PP_OCR("/root/models/pp_ocr.mud") SAVE_DIR = "/root/models/mymodel/" os.makedirs(SAVE_DIR, exist_ok=True) # 硬件初始化（假设已经设置好为RGB888） cam = camera.Camera(detector.input_width(), detector.input_height()) disp = display.Display() # 状态变量 last_save_time = 0 save_count = 0 def ocr_processing(img_path): """OCR处理函数""" try: # 加载图像 img = image.load(img_path, format=image.Format.FMT_BGR888) # 修改加载图像格式为BGR888 # OCR识别处理 objs = ocr.detect(img) text = ''.join([obj.char_str() for obj in objs]) # 可视化增强处理 for obj in objs: points = obj.box.to_list() img.draw_keypoints(points, image.COLOR_RED, 4, -1, 1) img.draw_string(obj.box.x4, obj.box.y4, obj.char_str(), image.COLOR_RED) disp.show(img) return text except Exception as e: print(f"[OCR异常] {str(e)}") return "" while not app.need_exit(): try: img = cam.read() objs = detector.detect(img, conf_th=0.5, iou_th=0.45) # 物体检测处理 if len(objs) > 0 and (time.time() - last_save_time) >= 3: filename = f"{SAVE_DIR}detect_{save_count}_{int(time.time())}.jpg" # 创建保存用图像副本并保存 save_img = img.copy() if save_img.save(filename, quality=85): print(f"[图像保存] {filename}") last_save_time = time.time() save_count += 1 # 将图像从RGB888转换为BGR888再进行OCR处理 converted_img = save_img.convert(image.Format.FMT_BGR888) converted_filename = f"{SAVE_DIR}converted_{save_count}_{int(time.time())}.jpg" converted_img.save(converted_filename, quality=85) ocr_result = ocr_processing(converted_filename) print(f"[识别日志] {ocr_result}") # 显示处理增强 for obj in objs: img.draw_rect(obj.x, obj.y, obj.w, obj.h, color=image.COLOR_RED) msg = f'{detector.labels[obj.class_id]}: {obj.score:.2f}' img.draw_string(obj.x, obj.y + 2, msg, scale=0.5, color=image.COLOR_RED) disp.show(img) except Exception as e: print(f"[运行异常] {str(e)}") break ``` ### 主要修改点： 1. **OCR 处理时的图像格式**：在 `ocr_processing` 函数中，将图像加载格式改为 `FMT_BGR888`。 2. **保存图像前的格式转换**：在保存图像用于 OCR 前，将图像从 `RGB888` 转换为 `BGR888` 并保存新的文件以供 OCR 使用。 这样可以确保图像格式与 OCR 模型的要求一致，避免格式不匹配的问题。