PaddleHub pyramidbox_lite_mobile使用gpu对视频进行推断出错

44u64gxh  于 4个月前  发布在  其他
关注(0)|答案(4)|浏览(36)

PaddleHub版本为2.3.1

PaddlePaddle-GPU版本为2.4,在AIStudio上GPU环境下运行

采用模型为paddlehub中pyramidbox_lite_mobile,并成功实现GPU对图像推断

后期二次开发中,对视频进行推断出错,报错为AttributeError: 'PyramidBoxLiteMobile' object has no attribute 'gpu_predictor'

经过查阅资料,可能是动态静态问题,请问这个问题应该如何解决呢,下面是ipynb文件

非常感谢您的指导!

ctehm74n

ctehm74n1#

我刚刚在aistudio上运行了你的这个文件,测试了单张图像是可行的。由于我没有视频素材,所以我用单张图像运行了一下process_frame函数,需要处理一些其他的错误。

mftmpeh8

mftmpeh82#

您好,这个问题可能是由于您的GPU不支持视频编码格式或者您的GPU性能不足导致的。您可以尝试更换视频编码格式或者升级您的GPU。

rqenqsqc

rqenqsqc3#

我刚把你的这个文件放到aistudio上跑了一下,测了单张图是可以的,视频我没有素材,然后我用单张图像跑了一下process_frame函数,有一些别的error需要处理一下。
链接: https://pan.baidu.com/s/1V1E4q1CsM4xWl8AeMQnQ5g 提取码: auc5
这是供您跑的一个视频,cpu已经能跑起来了,但是gpu不行,您说的error是哪些呢

sxpgvts3

sxpgvts34#

我在AI Studio里使用你的视频运行时没有出现上述错误。此外,请检查一下你的scalar变量名,有一个if分支下没有定义,还有一个位置变量名写错了。我修改了这个问题后,不再出现那个静态图的错误。

相关问题