PaddleHub PaddleX转换的模型,摄像头调用有问题

56lgkhnf  于 6个月前  发布在  其他
关注(0)|答案(3)|浏览(45)

paddleX转换的模型,在摄像头调用时出现问题。

调用图片没有问题,但是在使用cv2.VideoCapture(0)读取摄像头画面时,出现了错误提示:im应该是3维的,但现在是2维的。

翻译结果:

paddleX转换的模型在摄像头调用时出现问题。调用图片没有问题,但是在使用cv2.VideoCapture(0)读取摄像头画面时,出现了错误提示:im应该是3维的,但现在是2维的。

uklbhaso

uklbhaso1#

请先检查输入的图像形状和模型所需的图像形状是否匹配。如果不匹配,请查看使用的模型以及输入的frame是什么形状的图像。

4xrmg8kj

4xrmg8kj2#

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

x9ybnkn6

x9ybnkn63#

如果您使用的是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)。

相关问题