目录：
1.必要环境
2.YOLOX安装及模型推理
3.训练自己数据集及测试
1.必要环境
我的环境是：Windows10+python3.9+CUDA11.1+CUDNN8.1.1
在此环境配置不再多说，网上有很多的教程。本篇主要还是记录YOLOX的使用。（这里安装torch有个小坑，如果起初未安装torch,使用requirements.txt安装，则安装的为cpu版本，而此处我们使用的是gpu版本，注意别踩坑）
2.YOLOX安装
配置YOLOX的环境
2.2 安装依赖包：
pip install -r requirements.txt
2.3 安装yolox
python setup.py install
2.4 下载apex，解压到项目文件夹内 地址：mirrors / nvidia / apex · GIT CODE
2.5 cd到apex文件夹内，安装apex
cd apex-master
python setup.py install
2.6 返回yolox的主文件夹下，安装pycocotools
pip install pycocotools
2.7下载预训练模型，在2.1步骤给的网址中即可下载
2.8测试效果
测试图片指令：
python tools/demo.py image -f exps/default/yolox_s.py -c ./yolox_s.pth --path assets/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result --device gpu
效果：
3. 训练自己的数据集
此处制作数据集的步骤不再赘述。
3.1 首先按要求设置文件夹目录：
将图片放入JPEGImages中，xml文件放入Annotations
3.2 使用txt_write.py,划分数据集，并写入main文件夹下，生成trainval.txt和test.txt。
import os
import random
trainval_percent = 0.1
train_percent = 0.9
xmlfilepath = 'E:/YOLOX-main/datasets/VOC/VOCdevkit/VOC2007/Annotations'
txtsavepath = 'E:/YOLOX-main/datasets/VOC\VOCdevkit/VOC2007/ImageSets'
total_xml = os.listdir(xmlfilepath)
num = len(total_xml)
list = range(num)
tv = int(num * trainval_percent)
tr = int(tv * train_percent)
trainval = random.sample(list, tv)
train = random.sample(trainval, tr)
ftest = open('E:/YOLOX-main/datasets/VOC/VOCdevkit/VOC2007/ImageSets/Main/test.txt', 'w')
ftrain = open('E:/YOLOX-main/datasets/VOC/VOCdevkit/VOC2007/ImageSets/Main/trainval.txt', 'w')
for i in list:
name = total_xml[i][:-4] + '\n'
if i in trainval:
ftest.write(name)
else:
ftrain.write(name)
ftrain.close()
ftest.close()
3.3 修改 yolox/data/dataloading.py 如下：
3.4 修改exps/example/yolox_voc/yolox_voc_s.py如下：
修改num_classes改为自己的类别数量
3.5 修改yolox/data/datasets/voc_classes.py为自己的类别
3.6 修改yolox/evaluators/voc_eval.py 如下
在yolox/exp/yolox_base.py
修改self.data_num_workers = 0
执行python setup.py install 更新yolox
此时运行训练代码报如下错误：Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
方法解决：在train.py文件中添加代码：
import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
3.7 运行代码开始训练 ：
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 --fp16 -o -c yolox_s.pth
-d 使用显卡个数
-b 批次大小
–fp16 是否开启半精度训练
-c 加载预训练模型
如果出现CUDA out of memory内存溢出，将–fp16 -o去掉，代码改为如下：
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 -c yolox_s.pth
继续上一批次恢复训练的话，改成如下形式：
将—resume改为True，-c后面跟接要继续训练的权重文件
代码如下：
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 -c YOLOX_outputs/yolox_voc_s/latest_ckpt.pth
运行截图：
为了演示，本次训练，就训练了2轮。
结果存入：
测试：
4.1 测试前修改demo.py如下：
注释掉COCO_CLASSES,加入VOC_CLASSES
4.2 测试代码
python tools/demo.py image -f exps/example/yolox_voc/yolox_voc_s.py -c YOLOX_outputs/yolox_voc_s/latest_ckpt.pth --path assets/001.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result --device gpu
结果：
因为训练轮数为2轮，所以识别框不是很准确。
到此，YOLOX复现成功。
完结！
Original: https://blog.csdn.net/lc_MVP/article/details/121890017
Author: lc_MVP
Title: Windows10下搭建YOLOX环境以及训练自己数据集
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/682310/
转载文章受原作者版权保护。转载请注明原作者出处！