YOLOX3.0版本训练

YOLOX3.0版本训练教程

1. 训练配置

1. 自己git 代码，装环境：YOLOX

2. 准备VOC数据集，格式如下图
   
   其中VOC2007上面那两个文件夹是自动生成的不用管

3. 修改exps/example/yolox_voc/yolox_voc_s.py里的类型信息、数据集信息
   
   

4. 修改 yolox/data/datasets/voc_classes.py的类别，改为自己的类别

5. 修改yolox/data/datasets/voc.py中的这一行代码
   

6.重新编译：

python setup.py develop

2. 其他

• 若使用的VOC数据集进行训练，则不能使用wandb作为logger，因为目前官方代码还不支持。会报错'VOCDetection' object has no attribute 'cats
• 官方代码使用tensorboard 只能绘制mAP曲线，可以自己在 添加代码绘制loss曲线，学习率曲线等。
  
  添加代码如下：

                self.tblogger.add_scalar("lr", self.meter["lr"].latest, self.epoch + 1)
                loss_meter = self.meter.get_filtered_meter("loss")
                for k, v in loss_meter.items():
                    self.tblogger.add_scalar(k, v.latest, self.epoch + 1)

但是注意，这个代码不支持1个epoch验证一次，要从2开始。