#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
@Time        : 2022/4/25 10:30
@Author      : Albert Darren
@Contact     : 2563491540@qq.com
@File        : rename_dataset.py
@Version     : Version 1.0.0
@Description : TODO 将已有数据集转换组织方式，使得一个样本图片对应一个标签文件
@Created By  : PyCharm
"""
import os

root_dir = "hymenoptera_data/train"
label_dir = "bees_image"
label = label_dir.split("_")[0]
img_list = os.listdir(os.path.join(root_dir, label_dir))
output_dir = os.path.join(root_dir, label + "_label")
extension_name = ".txt"
for img_name in img_list:
    basename, *_ = img_name.rpartition(".")
    with open(os.path.join(output_dir, "{}{}".format(basename, extension_name)), "wt", encoding="utf-8") as fw:
        fw.write(label)