- 打开终端，确保已经到了 YOLOX 的目录下，若没有则
cd YOUR_YOLO_DIR，训练采用命令行的形式，一般采用fine turning的方式训练自己的数据集，命令如下
$ python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 -c weights/yolox_s.pth
- 如果出现
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.（笔者未碰到） 则在tools/train.py的开头中加入如下的代码：
import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
- 如果在训练了10个epoch后测试时找不到文件（笔者未碰到），则可以修改
./yolox/evaluators/voc_eval.py
...
def parse_rec(filename):
""" Parse a PASCAL VOC xml file """
tree = ET.parse(os.path.join(r"datasets/VOCdevkit/VOC2007/Annotations",filename))
objects = []
for obj in tree.findall("object"):
obj_struct = {}
obj_struct["name"] = obj.find("name").text
obj_struct["pose"] = obj.find("pose").text
obj_struct["truncated"] = int(obj.find("truncated").text)
obj_struct["difficult"] = int(obj.find("difficult").text)
bbox = obj.find("bndbox")
obj_struct["bbox"] = [
int(bbox.find("xmin").text),
int(bbox.find("ymin").text),
int(bbox.find("xmax").text),
int(bbox.find("ymax").text),
]
objects.append(obj_struct)
return objects
...
Original: https://blog.csdn.net/oliverwang15/article/details/124234417
Author: 兰心飞侠
Title: 利用 YOLOX 训练自己的目标检测数据集
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/681185/
转载文章受原作者版权保护。转载请注明原作者出处！