文章讲述了在yolo_traking-7.0的REID模块中，由于模型文件格式不兼容，作者如何通过torch库加载osnet_x0_25_msmt17.pth文件并将其转换为.pt格式，以避免复杂的代码改动。
yolo_traking-7.0中REID模块用到了osnet，track.py中模型文件不支持.pth，而model_zoo中仅有.pth，改动代码太麻烦了，网上查到的.pth文件转化为.pt文件都需要读取网络架构，不太可能实现。
读取osnet_x0_25_msmt17.pth发现是OrderedDict格式的，直接保存为.pt即可
import torch
model = torch.load('osnet_x0_25_msmt17.pth')
torch.save(model, 'osnet_x0_25_msmt17.pt')
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
