- 博客(3)
- 问答 (3)
- 收藏
- 关注
RSS订阅原创 百度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
251
原创 深度学习YOLOV5 笔记
训练好的pt转换onnxhttps://blog.csdn.net/chencaw/article/details/118057505
2021-12-13 08:47:06
834
原创 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
2860
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人