import json
import os
import shutil

#source_dir表示数据扩增之后的文件夹路径，此时标注的是多分类的标签
#new_dir表示转化之后得到的二分类文件夹


def to2class():
    #json存放路径
    source_dir = r'1'
    #json保存路径
    new_dir = r'1'
    for i in os.listdir(source_dir):
        if i.endswith('.json'):
            # json.load(open(path, encoding="utf-8", errors="ignore"))
            data = json.load(open(os.path.join(source_dir, i), encoding="utf-8", errors="ignore"))
            shapes = data['shapes']

            for item in shapes:
                # if item['label'] == '2':
                #     item['label'] ='1'  #! 负极标签变成2

                if item['label'] == '6':
                    item['label'] = '3'
                # if item['label'] == '5':
                #     item['label'] = '3'
                # else:
                #     item['label'] = '1' #! 正极标签变成1
            # json.dump(data, open(os.path.join(new_dir, i), 'w'))
            # json.dump(data, open(os.path.join(new_dir, i), 'w'))
            json.dump(data, open(os.path.join(new_dir, i),'w'),indent=3)
            print(i)
                    

if __name__=="__main__":
    to2class()