AI 22110 阅读 pytorch载入模型出现no module named models的解决办法 我目前在使用yolov5，它本身是用了pytorch。我在训练了数据集后，就把pt模型文件拷贝到其他项目中去了。可是在加载该pt文件时，却出现了类似'no module named models'的错误提示。在翻阅了资料后找到了原因和解决的办法之一。 假设训练的模型叫做m...... python 深度学习 YOLO pytorch AI
随记 16815 阅读 google colab的一些使用限制 目前有需求是训练自定义的数据集，用了yolov4 darknet，但是公司发的老版mac训练起来实在是慢且发烫，于是打算试一试google的colab，用一下免费的GPU。 先新建了一个notebook，选择菜单 代码执行程序-更改运行时类型-硬件加速器选择GPU。 然后在note...... YOLO
AI 9885 阅读 YOLOV5中使用pytorch的state dict来加载模型参数 前些时候在使用yolov5训练出模型但遇到了加载时提示no module named models的问题，当时用取巧的方式绕过了这个问题，让训练出的模型能在自己的项目中被成功加载，但当时的解决方式只是个临时的，以后当目录结构有变化时容易导致继续修改，于是看了yolov5的代码和pytorch的...... python 深度学习 YOLO pytorch AI
AI 工具框架 7301 阅读 pytorch转为onnx并调用时遇到的问题汇总 之前一直在用yolov5训练后生成的pytorch的模型文件，后缀名为pt，也用过pth用state dict加载处理，加载和推理用的代码较多，决定转换为onnx的类型用opencv加载，减少代码量也减少第三方包的安装，这样比较适合实际应用时部署。 state dict的记录在这里：YO...... python 深度学习 YOLO opencv pytorch
AI 7154 阅读 YOLOV5根据屏幕图像实时给出预测结果 目前在用yolov5，在调试时需要检验当前模型是否匹配出了期望的实体类型。因为目前需要匹配的就是当前计算机显示出的各类图像，于是在想做一个实时的检测屏幕图像并给出预测结果，就像之前文章中的根据摄像头实时给出预测一样。(不过有个问题，记录在后面) 除了yolov5需要的pytorch环境外...... python 深度学习 YOLO pytorch AI