paddleX转换的模型，摄像头调用有问题

[monkeycc]

paddleX转换的模型，摄像头调用有问题

result = module.predict(images=[(r"XXXX.jpg")]) 调用图片 没问题

但是 cv2.VideoCapture(0) ret, frame = cap.read() result = module.predict(images=frame,use_gpu=True)

摄像头调用的时候 报错 提示 Exception: im should be 3-dimensions, but now is 2-dimensions

[haoyuying]

用的是什么模型，查看下输入的frame是什么shape的图像，请先排查输入的图像的shape和模型需要的图像是否匹配。

[monkeycc]

frame.shape (480, 640, 3)

模型图像不知道哪里看 Model: FasterRCNN Transforms:

• Normalize: mean:
  • 0.485
  • 0.456
  • 0.406 std:
  • 0.229
  • 0.224
  • 0.225
• ResizeByShort: max_size: 1333 short_size: 800
• Padding: coarsest_stride: 32 target_size: null TransformsMode: RGB _Attributes: eval_metrics: bbox_map: 72.29197703403773 fixed_input_shape: null labels:
  • background
  • xx1
  • xx2 model_type: detector num_classes: 3 _ModelInputsOutputs: test_inputs:
  • • image
  • image
  • • im_info
  • im_info
  • • im_shape
  • im_shape test_outputs:
  • • bbox
  • multiclass_nms_0.tmp_0 _init_params: anchor_sizes:
  • 32
  • 64
  • 128
  • 256
  • 512 aspect_ratios:
  • 0.5
  • 1.0
  • 2.0 backbone: ResNet50 num_classes: 14 with_fpn: true completed_epochs: 0 status: Infer version: 1.2.8

[haoyuying]

若您使用的是paddlehub提供的模型，请给出具体的模型名称，目前您出错的问题应该是由于输入图像和模型要求的图像channel维度不匹配导致的，模型如果是您自己转的，可以查看下代码中input channel是多少。或者您可以参考paddlehub的检测模型的代码https://www.paddlepaddle.org.cn/hubdetail?name=faster_rcnn_resnet50_coco2017&en_category=ObjectDetection 尝试解决您的问题。 可以尝试result = module.predict(images=[frame],use_gpu=True)看能否解决您的问题。