
笔记
浮长川而忘返
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
百度AiStudio支持pytorch了!!!使用时遇到的bug
如果是rar文件,在pddlepaddle 2.3.2的环境有rar x name.rar可以解压文件,但在高级别的paddlepaddle版本没有rar命令。我在提供的pddlepaddle 2.3.2上试过,可以安装torch1.12.1且不会报错。后,import torch提示cuda有***11.so文件报错,可以尝试在控制台运行。自带的有tar和zip解压,所以最好压缩文件是tar或者zip。原创 2024-05-25 20:43:36 · 719 阅读 · 0 评论 -
深度学习YOLOV5 笔记
训练好的pt转换onnxhttps://blog.csdn.net/chencaw/article/details/118057505原创 2021-12-13 08:47:06 · 915 阅读 · 0 评论 -
pt模型在VS上运行
注意:本例子为我个人对YOLOv5的源码进行在VS上的实现,大家可按自己的模型根据源码进行修改我这为单个物体的检测,如果检测多个物体需对结果处理函数(non_max_suppression2)进行修改1.安装和训练时pytorch版本相对应的libtorch网上教程很多,我两个版本都是CUDA版本2.将训练好的模型转为.torchscript.pt模型(可以选择是CUDA或者CPU,这里是CUDA)import argparseimport torchimport torch.nn as n原创 2021-12-12 15:52:36 · 3268 阅读 · 0 评论