加载预训练模型出错

PeterH0323 / Smart_Construction

Base on YOLOv5 Head Person Helmet Detection on Construction Sites，基于目标检测工地安全帽和禁入危险区域识别系统，🚀😆附 YOLOv5 训练自己的数据集超详细教程🚀😆2021.3新增可视化界面❗❗

GNU General Public License v3.0

2.24k stars 469 forks source link

加载预训练模型出错 #6

Closed DL-Alva closed 4 years ago

DL-Alva commented 4 years ago

您好，我在训练过程中加载预训练模型时，报错：RuntimeError: ./weights/yolov5s.pt is a zip archive (did you mean to use torch.jit.load()? 运行环境是cuda:10.0, pytorch:1.5.1, 请问这是怎么回事呢？

PeterH0323 commented 4 years ago

我在另外一个 Issue 回答您，这个先关闭了

eeric commented 3 years ago

标签制作好，可以直接用https://github.com/ultralytics/yolov5 训练