Pytorch_Yolov5_Inference

Pytorch_Yolov5_Inference copied to clipboard

fix:添加map和cmath头文件

LiuZengqiang

yolov5s_1.3a.pt 这个文件是怎么获取到的？

3

由于ultralytics/yolov5 使用的是高版本的pytorch，训练得到的网络模型文件 是zip格式的，与该项目的pytorch不兼容 尝试使用以下代码重新保存权重文件： model = Model(opt.cfg).to(device) w = 'weights/yolov5s.pt' #来自官方训练权重，v3.0版本 model.load_state_dict(torch.load(w), False) torch.save(model.state_dict(), 'old_version.pt', _use_new_zipfile_serialization=False) 得到的old_version.pt后，使用该项目的代码执行： 1、python quant.py 能生成yolov5s_int8.pt 2、python detect.py 无报错，但检测不出任何目标 请问 问题在哪？该项目的 yolov5s_1.3a.pt 是怎么获取到的？ 期待您的回复。

qiqihuang

运行python detect.py --jit True --save True命令，生成yolov5.cambricon时提示RuntimeError: No such operator torch_mlu::yolov5_detection_output。需要怎么操作？

xyt2008

Aruen24