复盘自用：使用yolox训练自己的数据集（仅参考）

橘子果酱278

已于 2024-05-12 13:05:32 修改

阅读量519

点赞数 4

文章标签：深度学习人工智能

于 2024-04-21 20:53:23 首次发布

本文链接：https://blog.csdn.net/weixin_62906471/article/details/138044206

版权

本文详细指导如何在Python中安装YOLOX进行物体检测，包括解决PyTorch版本与CUDA不兼容问题，构建VOC格式数据集，修改配置文件以适应特定类别，以及在训练过程中遇到的问题和解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一安装环境

超详细！手把手教你使用YOLOX进行物体检测（附数据集） - 知乎 (zhihu.com)

参考这篇文章，装好先demo，demo成功说明没有问题。最好使用gpu推理，因为后期训练也是使用gpu。这样才能测试环境。

tips:我遇到的问题：pytorch capability sm_86 is not compatible with the current PyTorch installation。

torch版本与cuda版本不兼容。此处显卡为3090，我装了1.7.1和cuda11.0解决了问题。

直接执行这条命令解决了这个问题

pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/cu110/torch_stable.html

二构建voc格式数据集

在根目录下创建VOCdevkit/VOC2007

其中annotation存放xml标签文件，JPEGimage放原始图片。ImageSets/Main下的两个文件可以根据代码建立，创建代码参考这篇文章。

三修改配置文件

参考YOLOX训练自己的数据集(超详细)-CSDN博客

3.1类别定义

将里面的类别改为自己的类别名字

3.2 类别数量

将数字改为自己的类别数量

3.3数据文件路径

修改 yolox/data/dataloading.py改为自己的yolox路径

终于可以训练了！

 python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 --fp16  -c yolox_s.pth

更新：另一个训练方式

python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 16 --fp16 -o -c ./yolox_s.pth

3.4自己遇到的问题

3.4.1 路径导航不对

原因：前面在yolox执行了这段代码如下图，装了一个yolox的包，执行项目路径错误导航到这个包。解决办法：pip uninstall yolox，卸载掉

python setup.py install