HRNet 可视化python visualization/plot_coco.py

who are you605

于 2022-11-18 20:55:56 发布

阅读量704

点赞数

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.csdn.net/hdyan1118/article/details/125946039

本文介绍如何使用deep-high-resolution-net.pytorch项目进行COCO数据集上的人体姿态估计，并展示了如何运行可视化脚本来输出关键点结果，同时提供了解决常见错误的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

该代码输出的是COCO数据集中图片的人体姿态估计结果（只输出关键点）
一、下载代码
https://github.com/leoxiaobin/deep-high-resolution-net.pytorch

二、运行可视化文件

python visualization/plot_coco.py --prediction output/coco/CBAM_pose_hrnet/CBAM_w32_256x192_adam_lr1e-3/results/keypoints_val2017_results_0.json

三、报错解决

AttributeError: 'NoneType' object has no attribute 'shape'

解决方法：查看图片加载路径是否有误，修改为正确的路径。
在这里插入图片描述

    parser.add_argument('--image-path',
                        help='Path of COCO val images',
                        type=str,
#                         default='data/coco/images/val2017/'
                        default='data/coco/val2017/'
                        )