AI 22137 阅读 pytorch载入模型出现no module named models的解决办法 我目前在使用yolov5，它本身是用了pytorch。我在训练了数据集后，就把pt模型文件拷贝到其他项目中去了。可是在加载该pt文件时，却出现了类似'no module named models'的错误提示。在翻阅了资料后找到了原因和解决的办法之一。 假设训练的模型叫做m...... python 深度学习 YOLO pytorch AI
AI 9911 阅读 YOLOV5中使用pytorch的state dict来加载模型参数 前些时候在使用yolov5训练出模型但遇到了加载时提示no module named models的问题，当时用取巧的方式绕过了这个问题，让训练出的模型能在自己的项目中被成功加载，但当时的解决方式只是个临时的，以后当目录结构有变化时容易导致继续修改，于是看了yolov5的代码和pytorch的...... python 深度学习 YOLO pytorch AI
AI 工具框架 7323 阅读 pytorch转为onnx并调用时遇到的问题汇总 之前一直在用yolov5训练后生成的pytorch的模型文件，后缀名为pt，也用过pth用state dict加载处理，加载和推理用的代码较多，决定转换为onnx的类型用opencv加载，减少代码量也减少第三方包的安装，这样比较适合实际应用时部署。 state dict的记录在这里：YO...... python 深度学习 YOLO opencv pytorch
AI 7169 阅读 YOLOV5根据屏幕图像实时给出预测结果 目前在用yolov5，在调试时需要检验当前模型是否匹配出了期望的实体类型。因为目前需要匹配的就是当前计算机显示出的各类图像，于是在想做一个实时的检测屏幕图像并给出预测结果，就像之前文章中的根据摄像头实时给出预测一样。(不过有个问题，记录在后面) 除了yolov5需要的pytorch环境外...... python 深度学习 YOLO pytorch AI
AI 3122 阅读 解决问题PytorchStreamReader failed reading zip archive 在用到一个第三方的pth模型时遇到了一个奇怪的错误，加载模型在mac和linux一切正常，但在windows上加载时就遇到了报错，报错信息较长，但最后一个提示是出在pytorch的serailization里，如下： .... venv\lib\site-packages\torc...... python pytorch