如何在我们自己的数据集上训练yolov7模型？

Question

当我尝试通过执行下面的代码单元来训练我的模型时：

python train.py --img-size 2048 --cfg cfg/training/yolov7.yaml --hyp data/road_sign_data.yaml --batch 8 --epochs 100 --data data/road_sign.yaml --weights yolov7_training。 pt --workers 24 --name yolo_road_det

我有以下错误消息：

回溯（最近一次调用最后一次）： 文件“C:\UsersY558\Documents\streamline2\yolov7 rain.py”，第 12 行，位于 导入 torch.distributed as dist 文件“C：\ UsersY558 \ AppData \ Local \ Packages \ PythonSoftwareFoundation.Python.3.11_qbz5n2kfra8p0 \ LocalCache \ local-packages \ Python311 \ site-packages orch_init_.py”，第141行，位于 引发错误 OSError: [WinError 126] 找不到指定的模块。加载“C:\UsersY558\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.11_qbz5n2kfra8p0\LocalCache\local-packages\Python311\site-packages orch\lib\shm.dll”或其依赖项之一时出错。

看起来无法导入torch.distibuted

我尝试更改我正在使用的python版本（从3.12到3.11.9），但它仍然不起作用。我还尝试了许多其他方法来训练 yolov7 模型，但它们都不起作用......如果您有任何解决方案，那将非常有帮助

Answer 1

您需要在系统上安装 pytorch 并检查它是否正常工作：

https://pytorch.org/

^^ 向下滚动并选择您正在使用的适当系统，pip 或 conda、windows 或 mac 或 linux、Cuda 或其他，然后复制/粘贴它为您提供的命令来安装 torch/pytorch 和相关库。

可以通过运行检查安装是否成功：

python -c "import torch; print(torch.__version__)"

应该看到类似的输出：

2.3.0+cu118

或者你拥有的任何版本的 pytorch...如果有 nvidia 卡，肯定会获得 CUDA 支持。