- 博客(3)
- 收藏
- 关注
RSS订阅原创 torch的版本跟cuda不对应导致的数据集训练过慢
首先去官网下载对应的.whl文件，由于我是在服务器训练，所以下载的是linux版本。把下载好的文件放到文件夹里，我这里直接放进了工程文件夹，然后cd到文件夹所在路径。在服务器上训练yolov5的数据集，2080ti显卡训练一天发现只训练了十几轮。在\runs\train\exp中打开result可看训练的轮数。猜测可能是torch的版本跟cuda不对应导致的，下载新的torch。可以pip安装，但是太慢，网上有很多教程，这里介绍其中一种。安装完再进行训练发现速度提升，查看显卡也在使用。
2023-07-06 17:30:39
232
1
原创 YOLOv5-6.0保存识别结果到txt文件
目标检测在实际应用中可能会用到检测到的物体标签，这时需要我们读取检测到的标签文本。yolo本身自带的保存功能，但是保存的是yolo格式，只有0，1，2无法判断具体是什么。第一步，先定义一个变量，以便给文本命名。第二步，识别一张后变量+1。第三步，将结果存入文件夹。
2023-04-19 22:17:32
2333
1
原创 把python文件转为exe可执行文件
若执行pyinstaller --onefile love.py时候遇到报错：The 'typing' package is an obsolete backport of a standard library package and is incompatible with PyInstaller. Please `conda remove typing` then try again.库，而该库已经成为 Python 的标准库之一，为了避免上述错误，您可以在执行。命令前，执行以下命令卸载。
2023-04-02 12:16:45
693
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人