【MMPose】在HRNet应用SimDR(SimCC)/Part.1-配置文件篇

文章介绍了如何在MMPose框架中基于HRNet实现SimDR(现称SimCC),这是一种将姿态估计的Heatmap方法转换为分类的方法,能提高效率并减少显存占用。作者详细阐述了在配置文件、数据预处理流水线、检测头以及损失函数等方面的修改步骤,包括新增参数、修改检测器类型、替换检测头和损失函数以及调整数据预处理流程。

 github代码已经上传:mmpose_simDR


SimDR(现在已经改名叫SimCC,后文还是称SimDR)将姿态估计的Heatmap方法转换为分类方法,在HRNet上实现了涨点,并且减小了显存占用。作者已经在github上开源了代码,但是在MMPose上目前还没有实现,所以本篇文章就使用HRNet在MMPose上实现SimDR。

SimDR原文: Is 2D Heatmap Representation Even Necessary for Human Pose Estimation?

SimDR开源代码:SimCC

因为在MMPose上修改的部分较多,所以文章会分为以下几个部分:

  1. 配置文件篇:因为MMPose使用了配置文件(.py)来进行各种参数的设置,所以我们先将会用到的参数先在配置文件中给出,方便后面修改代码时的调用与理解;
<think>我们正在解决的是:在Windows系统中下载ANIMAL-POSE数据集(具体文件:https://download.openmmlab.com/mmpose/datasets/animalpose.tar)的问题。用户提到wget在Windows下不可用,需要替代方案。参考引用:引用[1]和[2]中使用了wget命令,但这是在Linux/macOS环境下的方法。引用[2]还提到了使用MMPose提供的数据集下载链接。解决方案:1.在Windows系统中,我们可以使用其他工具替代wget,例如:-使用浏览器直接下载-使用PowerShell的Invoke-WebRequest命令(类似于wget)-安装Windows版本的wget-使用其他下载工具(如curl、aria2等)2.具体到下载链接:https://download.openmmlab.com/mmpose/datasets/animalpose.tar步骤建议:方法一:浏览器直接下载直接复制链接到浏览器地址栏,按回车即可下载。方法二:使用PowerShell(Windows自带)打开PowerShell(按Win+R,输入powershell,回车),然后执行:Invoke-WebRequest-Uri"https://download.openmmlab.com/mmpose/datasets/animalpose.tar"-OutFile"animalpose.tar"方法三:安装wgetforWindows1.下载wgetforWindows:https://eternallybored.org/misc/wget/2.将下载的wget.exe放到系统路径(如C:\Windows\System32)或任意目录(然后将该目录添加到PATH环境变量)3.打开命令提示符(cmd)或PowerShell,执行:wgethttps://download.openmmlab.com/mmpose/datasets/animalpose.tar-Oanimalpose.tar方法四:使用curl(Windows10及以后版本自带curl)在PowerShell或命令提示符中执行:curl-oanimalpose.tarhttps://download.openmmlab.com/mmpose/datasets/animalpose.tar注意:由于文件较大(约1.2GB),下载可能需要较长时间,请确保网络稳定。下载后的处理:下载得到的是tar归档文件,在Windows中可以使用解压工具(如7-Zip、WinRAR等)解压。建议:如果用户后续需要在MMPose中使用该数据集,请按照之前回答中的目录结构放置。引用说明:我们参考了引用[2]中的下载链接,该链接是MMPose官方提供的预处理后的ANIMAL-POSE数据集。生成相关问题:根据用户当前的问题(Windows下下载数据集)和之前的问题(关于ANIMAL-POSE数据集),可以生成以下相关问题:1.如何在Windows中解压.tar格式的ANIMAL-POSE数据集文件?2.如何将下载的ANIMAL-POSE数据集正确配置到MMPose训练环境中?3.ANIMAL-POSE数据集在MMPose中的默认目录结构是什么?4.如果下载过程中断,如何继续下载?注意:在回答中,我们不需要列出引用,因为用户提供的引用只是背景信息,我们实际使用的是MMPose的下载链接(引用[2]),但无需在回答中标注引用标记,因为用户没有要求引用格式,且问题集中在技术操作上。</think>### 在 Windows 系统中下载 ANIMAL-POSE 数据集的解决方案 针对 Windows 系统无法直接使用 `wget` 命令的问题,以下是四种可靠的方法下载数据集([https://download.openmmlab.com/mmpose/datasets/animalpose.tar](https://download.openmmlab.com/mmpose/datasets/animalpose.tar)): #### 方法 1:使用 PowerShell 命令(无需安装) ```powershell # 打开 PowerShell (Win+X → Windows PowerShell) Invoke-WebRequest -Uri "https://download.openmmlab.com/mmpose/datasets/animalpose.tar" -OutFile "animalpose.tar" ``` > ✔️ **优点**:Windows 系统自带,无需额外安装 > ⚠️ **注意**:大文件下载建议添加 `-Resume` 参数支持断点续传 #### 方法 2:安装 Windows 版 wget 1. 下载 wget for Windows: [https://eternallybored.org/misc/wget/](https://eternallybored.org/misc/wget/) 2. 解压后将 `wget.exe` 放入 `C:\Windows\System32` 3. 在命令提示符执行: ```cmd wget https://download.openmmlab.com/mmpose/datasets/animalpose.tar -O animalpose.tar ``` #### 方法 3:使用浏览器直接下载 1. 复制下载链接到浏览器地址栏: `https://download.openmmlab.com/mmpose/datasets/animalpose.tar` 2. 浏览器将自动开始下载(约 1.2GB) 3. 保存为 `animalpose.tar` #### 方法 4:使用 cURL(Windows 10+ 内置) ```cmd curl -o animalpose.tar https://download.openmmlab.com/mmpose/datasets/animalpose.tar ``` ### 数据集解压说明 下载完成后,使用以下工具解压: 1. **7-Zip**(推荐):[https://www.7-zip.org/](https://www.7-zip.org/) 2. **WinRAR**:[https://www.win-rar.com/](https://www.win-rar.com/) ```powershell # 使用 PowerShell 解压 tar -xvf animalpose.tar -C 目标目录 ``` ### MMPose 数据集配置建议 解压后目录结构应为: ``` animalpose/ ├── annotations/ │ ├── animalpose_train.json │ └── animalpose_val.json └── images/ ├── cat/ ├── dog/ ├── horse/ ├── sheep/ └── cow/ ``` 在 MMPose 配置文件中修改数据集路径: ```python data_root = 'data/animalpose/' train_data_cfg = dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_train.json', img_prefix=f'{data_root}/images/') ```
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值