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')