行人重识别网络模型训练

Question

有人用过Github的这个项目吗：https://github.com/KaiyangZhou/deep-person-reid
用来训练行人重识别网络模型的。
我在自己电脑上确实成功复现了，得到了model.pth.tar-1这样的文件。

但是我是希望通过这个deep-person-reid项目训练得到权重文件能应用到YOLO_Tracking中去。
官方确实给了不少权重文件：osnet_x0_25_market1501.pth osnet_x0_25_msmt17.pth osnet_x1_0_msmt17.pth
但是我自己训练却始终无法得到这种格式的文件，只有tar文件。
想要到他的官网去下载这些权重文件也无法下载，网页无法显示。

Answer 1

让阿豪来帮你解答，本回答参考chatgpt3.5编写提供，如果还有疑问可以评论或留言

问题：如何将通过deep-person-reid项目训练得到的model.pth.tar文件转换成官方提供的以.pth结尾的权重文件？ 回答：

1. 首先需要明确，.tar文件是压缩文件，包含了.pth文件和其他文件，我们需要先将.pth文件提取出来。
2. 提取.pth文件的方法有多种，可以使用WinRAR、7-ZIP等压缩软件解压缩，也可以使用Python代码进行解压缩。 以Python代码为例，在命令行或Python环境下输入以下代码，将.tar压缩文件内的.pth文件提取到指定目录下。

   import tarfile
   tar = tarfile.open('model.pth.tar', 'r')
   tar.extract('model.pth', path='指定的输出目录')
   tar.close()
   

   注意，这里提取的是model.pth文件，而非model.pth.tar文件。

3. 接下来需要将.pth文件转换成官方提供的.pth格式。 首先需要明确，deep-person-reid项目使用的是PyTorch框架，而官方提供的权重文件也是以PyTorch格式保存的。因此，我们只需要将.pth文件直接改名为官方提供的权重文件名，即可使用官方的权重文件。 例如，将model.pth重命名为osnet_x0_25_market1501.pth。

   mv model.pth osnet_x0_25_market1501.pth
   

4. 如果无法从官网下载官方的权重文件，可以尝试从其他渠道下载，或者联系官方获取。 注意，使用未知来源的权重文件存在安全风险，请谨慎使用。 举例： 已经在deep-person-reid项目中训练得到了model.pth.tar文件，现在希望将其转换成官方提供的osnet_x0_25_market1501.pth权重文件。
5. 使用Python代码将.pth文件提取出来，保存到当前目录下：

   import tarfile
   tar = tarfile.open('model.pth.tar', 'r')
   tar.extract('model.pth', path='./')
   tar.close()
   

6. 将提取出来的.pth文件重命名为osnet_x0_25_market1501.pth：

   mv model.pth osnet_x0_25_market1501.pth
   

现在，已经可以将osnet_x0_25_market1501.pth权重文件应用到YOLO_Tracking中去。