深圳做网站建设的公司,鄂北局网站建设者风采,如何跟客户销售做网站,优化网站的公司yolo_traking-7.0中REID模块用到了osnet#xff0c;track.py中模型文件不支持.pth#xff0c;而model_zoo中仅有.pth#xff0c;改动代码太麻烦了#xff0c;网上查到的.pth文件转化为.pt文件都需要读取网络架构#xff0c;不太可能实现。 读取osnet_x0_25_msmt17.pth发现…yolo_traking-7.0中REID模块用到了osnettrack.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)