【YOLO-Pose】在Windows上的部署与测试(调用摄像头)

文章介绍了如何解决在PyTorch环境中遇到的Upsample对象无recompute_scale_factor属性的报错,并详细提供了修改源码的步骤。接着,对比了HRNet和YOLO-Pose在多人姿势估计上的差异,指出YOLO-Pose因端到端的网络结构和较快的推理速度适用于低算力设备。文章提供了YOLO-Pose的部署环境(Windows11,Python3.7,PyTorch1.13,CUDA11.6)和源码获取、环境配置、模型安装及测试的详细过程。

2023.4.20更新 关于报错的解决办法

关于评论出现的问题,我当初测试的时候是没有的(可能是以前跑YOLO的时候已经改过了),于是我换了一台电脑测试,报错了,于是这里更新一下:

报错'Upsample' object has no attribute 'recompute_scale_factor'的解决办法:

找到你的pytorch源码中的upsampling.py文件(我的位置是在anaconda3\envs\torch12\lib\site-packages\torch\nn\modules\upsampling.py),将153行开始的forward函数做修改:

def forward(self, input: Tensor) -> Tensor:
        # return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
        #                      recompute_scale_factor=self.recompute_scale_factor)
        return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)

这样再根据下文的的说明来操作就有结果了。


虽然基于top-down方法的HRNet网络精度很高(在COCO数据集上最高能达到76.7mAP),但是由于其需要一个额外的detector和网络中始终保持着高分辨率的heatmap的原因,网络在推理时的速度非常的慢,这就使得类似与HRNet的网络不适合部署在低算力的边缘设备上。

而使用buttom-up的方法虽然快,但是其精度与top-down方法还存在一些差距。

YOLO-Pose与其他buttom-up的方法一样,也是一种single shot的方

评论 10
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值