代码是依照自带的yolov5的代码换了模型和data
KeyError                                  Traceback (most recent call last)
/tmp/ipykernel_106356/2101816269.py in <cell line: 3>()
      5     infer_image(img_path, model, labels_dict, cfg)
      6 elif infer_mode == 'camera':
----> 7     infer_camera(model, labels_dict, cfg)
      8 elif infer_mode == 'video':
      9     video_path = 'racing.mp4'

/tmp/ipykernel_106356/2968311708.py in infer_camera(model, labels_dict, cfg)
    123         # 对摄像头每一帧进行推理和可视化
    124         _, img_frame = cap.read()
--> 125         image_pred = infer_frame_with_vis(img_frame, model, labels_dict, cfg)
    126         image_widget.value = img2bytes(image_pred)

/tmp/ipykernel_106356/2968311708.py in infer_frame_with_vis(image, model, labels_dict, cfg, bgr2rgb)
     75     scale_coords(cfg['input_shape'], pred_all[:, :4], image.shape, ratio_pad=(scale_ratio, pad_size))
     76     # 图片预测结果可视化
---> 77     img_vis = draw_bbox(pred_all, image, (0, 255, 0), 2, labels_dict)
     78     return img_vis
     79 

/tmp/ipykernel_106356/2968311708.py in draw_bbox(bbox, img0, color, wt, names)
     17         img0 = cv2.rectangle(img0, (int(bbox[idx][0]), int(bbox[idx][1])), (int(bbox[idx][2]), int(bbox[idx][3])),
     18                              color, wt)
---> 19         img0 = cv2.putText(img0, str(idx) + ' ' + names[int(class_id)], (int(bbox[idx][0]), int(bbox[idx][1] + 16)),
     20                            cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1)
     21         img0 = cv2.putText(img0, '{:.4f}'.format(bbox[idx][4]), (int(bbox[idx][0]), int(bbox[idx][1] + 32)),

KeyError: 7995