训练FastestDet（Anchor-Free、参数量仅0.24M、计算量0.9GFLOPs），稍改代码使得符合YOLO数据集排布

0 参考链接

• 官方的代码：FastestDet

1 准备数据

我已有的数据集排布：（符合YOLO排布）

dataset:.
├─images
│  ├─test
│  │  ├─xxx.jpg
│  │  ├─xxx.jpg
│  ├─train
│  └─val
├─labels
│  ├─test
│  │  ├─xxx.txt
│  │  ├─xxx.txt
│  ├─train
│  └─val

1.1 使用以下代码生成绝对路径的txt文件

• root_dir：数据集根目录
• save_dir：存储目录
• 结果：生成3个txt文件abspath_xxx.txt

import os

if __name__ == '__main__':
    root_dir = r'F:/A_Publicdatasets/RDD2020-1202/train_valid/RDD2020_together/images'
    save_dir = 'F:/A_Publicdatasets/RDD2020-1202/train_valid/RDD2020_together/'

    for s in ['train', 'val', 'test']:
        save_path = f'{
     
     save_dir}/abspath_{
     
     s}.txt'
        with open(save_path, 'w') as f:
            for