18 lines
452 B
Python
18 lines
452 B
Python
from create_extra import split_subdirs
|
|
from data_split import split_directory
|
|
from extend import ImageExtendProcessor
|
|
import yaml
|
|
|
|
|
|
def data_preprocessing(conf):
|
|
# split_subdirs(conf)
|
|
# image_ex = ImageExtendProcessor(conf)
|
|
# image_ex.control_number()
|
|
split_directory(conf)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
with open('../configs/sub_data.yml', 'r') as f:
|
|
conf = yaml.load(f, Loader=yaml.FullLoader)
|
|
data_preprocessing(conf)
|