大漠Yolo训练失败，一直失败。请问有人会解决吗？电脑配置

MrKiss

独特舞步 发表于 2024-8-30 11:14
配置文件是这样设置的吗？我只设置了这几个而已。

环境变量是在系统的Path里面设置 python的目录和 python的插件目录
比如 安装的地方是E盘：e:\python312\
                      插件路径：e:\python312\Scripts\
然后在 系统变量 创建 变量
变量名                                             变量值
PYTHONDONTWRITEBYTECODE         1

独特舞步

MrKiss 发表于 2024-8-30 19:40
环境变量是在系统的Path里面设置 python的目录和 python的插件目录
比如 安装的地方是E盘：e:\python312\ ...

是这样吗？我试了，出现 也是不行。哎，请问能把你的环境变量截图给我看看吗？

独特舞步

MrKiss 发表于 2024-8-29 13:39
你是进行那一步的时候 出现了这个

请问是不是要安装"Anaconda"?我没有安装，我下载了好像不知道在哪配置YOLO。。。

MrKiss

独特舞步 发表于 2024-8-30 21:06
是这样吗？我试了，出现也是不行。哎，请问能把你的环境变量截图给我看看吗？
...

你出现这个情况是 测试训练的时候出现的吗? anaconda不用安装也可以的

补充内容 (2024-8-31 17:16):
你要看清楚  测试yolo的时候 是在 yolov5-7.0 文件夹里面调试cmd  

独特舞步

MrKiss 发表于 2024-8-31 17:14
你出现这个情况是 测试训练的时候出现的吗? anaconda不用安装也可以的

补充内容 (2024-8-31 17:16):

测试过了正常，但是在YOLO5.7.0目录下没有发现有yolov5s.onnx。大漠配置文件里说会出现的。现在没有出现。

独特舞步

MrKiss 发表于 2024-8-31 17:14
你出现这个情况是 测试训练的时候出现的吗? anaconda不用安装也可以的

补充内容 (2024-8-31 17:16):

哪里安装错了吗？需求环境(只是训练的机器环境,用插件的环境没有任何要求)
1. win10或者win11 64位系统. 其他系统没测试过,不一定能行. 系统最好最新版本. (必须)
2. 最好有nvidia显卡,并且显卡驱动更新到最新的版本,最好是30系列以上. 如果没有的话,训练效率会很低下. (可选,但是还是强烈建议)
3. 全局科学上网.因为很多安装包都需要到wai网下载.(必须)




安装步骤
1.  安装python 3.8以上的版本.  
    下载地址 https://www.python.org/downloads/
    安装时,记得勾选添加路径到系统环境变量.其他可以默认. 如果忘记了勾选添加路径到环境变量,那需要手动添加环境路径.
    方法是打开windows环境变量,在系统变量的Path中添加python的路径. 比如我这里是
    C:\Users\Administrator\AppData\Local\Programs\Python\Python311
    C:\Users\Administrator\AppData\Local\Programs\Python\Python311\Scripts
    你自己安装时,会稍有不同,根据你的系统用户名和python的版本会不同.
    另外最好自己再手动加一个环境变量,避免执行py脚本时,产生cache文件夹
    PYTHONDONTWRITEBYTECODE=1


2. 安装cuda (如果需求环境的条件2不满足,那这个步骤可以略过)
    下载地址 https://developer.nvidia.com/cuda-downloads
    Operating System 选windows.  Architecture选 x86_64.  Version win10和win11对应的10和11版本. Installer Type选exe(local)
    安装要选择自定义安装,并且只勾选cuda. 其他一律不选(重要!!!)
   
3. 安装PyTorch.  
    下载地址 https://pytorch.org/get-started/locally  
    PyTorch Build选Stable版本.  Your OS选windows. Package 选Pip. Language选Python.
    Compute Platform选最新的cuda.比如我这里是cuda11.8(如果需求环境的条件2不满足,那么这里选CPU)
    之后下面的Run this Command会生成一个命令. 复制这个命令,打开命令行粘贴进行运行即可.
    比如我这里是pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118


4. 安装yolo  注意yolo路径里,不要有任何非英文字符.
    直接用我下载好的yolo.rar即可. 放在任意目录. 建议放根目录. 我这里是放在了e盘.
    解压yolo.rar到E盘根目录. 解压后的目录形式是这样的 e:/yolo/yolov5_7.0
    命令行进入到yolov5_7.0这个目录下,然后执行以下命令
    pip install -r requirements.txt


5. 安装git
    下载地址 https://git-scm.com/download/win
     选择Standalone Installer下的64位安装包


6. 测试识别
    命令行进入yolov5_7.0目录,然后执行以下命令
    python detect.py --weights yolov5s.pt --source data/images/bus.jpg
    最后会在runs/detect目录下生成识别后的图像.


7. 测试训练 使用coco128来测试
    命令行进入yolov5_7.0目录,然后执行以下命令
    python train.py --weights yolov5s.pt --epochs 300 --batch-size 16 --workers 8 --data ../datasets/coco128/coco128.yaml
    成功后,会在runs/train目录下生成训练后的模型和标记的图片等.


8. 测试模型格式转换 使用yolov5s.pt来测试
    命令行进入yolov5_7.0目录,然后执行以下命令
    python export.py --weights yolov5s.pt --simplify --include onnx
        成功后,会在yolov5_7.0目录下生成yolov5s.onnx文件.
       
至此,说明你的环境已经配置完成. 其他的请观看我的视频教程. 后续会一步一步教你如何搭配插件来使用.

我都全部安装了都还是不行。训练失败一直都在！

MrKiss

独特舞步 发表于 2024-8-31 22:40
哪里安装错了吗？需求环境(只是训练的机器环境,用插件的环境没有任何要求)
1. win10或者win11 64位系统.  ...

6. 测试识别
    命令行进入yolov5_7.0目录,然后执行以下命令
    python detect.py --weights yolov5s.pt --source data/images/bus.jpg
    最后会在runs/detect目录下生成识别后的图像.
你尝试这一步后 报错的是什么

独特舞步

MrKiss 发表于 2024-9-1 01:52
6. 测试识别
    命令行进入yolov5_7.0目录,然后执行以下命令
    python detect.py --weights yolov5s.p ...

Traceback (most recent call last):  File "E:\yolo\yolov5-7.0\detect.py", line 35, in <module>
    import torch
ModuleNotFoundError: No module named 'torch'


就是这个错误！

MrKiss

独特舞步 发表于 2024-9-1 07:00
Traceback (most recent call last):  File "E:\yolo\yolov5-7.0\detect.py", line 35, in
    import t ...

缺少什么模块 那就 cmd 输入pip install -i https://mirrors.aliyun.com/pypi/simple/ torch

补充内容 (2024-9-1 15:00):
然后 在按照刚才的测试 在yolo文件夹打开CMD 输入测试代码 在看看缺少什么模块

独特舞步

MrKiss 发表于 2024-9-1 14:59
缺少什么模块 那就 cmd 输入pip install -i https://mirrors.aliyun.com/pypi/simple/ torch

补充内容 ( ...

不是缺少，应该是错误 35行