import torch
import torchvision.models
import sys
import pandas as pd
import os


abs_path = os.path.dirname(os.path.abspath(__file__))

#증식 하기 전의 데이터셋의 폴더 경로(상대경로)를 적어주세요
ds_root = 'raw_ds_201118_1.0'



print(abs_path)
for root, dirs, files in os.walk(ds_root, topdown = True):
    if '_a' in root:
        if 'train' in root:
            pass
        elif 'validation' in root:
            pass
        else:            
            prefix_id = root[-2:]            
            #print(prefix_id)
        for file_name in files:
            path = os.path.join(abs_path,root)
            print(path, file_name)
            new_file_name = prefix_id + '_' + file_name
            if prefix_id in file_name:
                pass
            else:
                os.rename(os.path.join(path+'\\'+file_name),os.path.join(path+'\\'+new_file_name))