JiangHejun / tvlogo_dataset

生成符合yolov5训练的格式的台标数据集

本项目主要用于为yolov5-pytorch生成台标数据集

• 支持多台标生成
• 图像缩放
• 图像加噪：椒盐、高斯
• 图像旋转
• 台标抠图
• 在运行python3 build.py之前，请先运行python3 build.py --finetune进行微调
• ./dataset/background和./dataset/tvlogos的目录结构固定，不可改变；增加新的标签（签名）请在./dataset/tvlogos目录之下新建对应的文件夹

.
├── README.md
├── build.py
└── dataset
    ├── background  // 背景图文件夹
    ├── fakes       // fake台标文件夹
    └── tvlogos     // 目标台标文件夹
        ├── chengdu
        ├── hunan
        └── jiangsu