| 1234567891011121314151617181920212223242526272829303132 |
- 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))
|