1 # -*- coding: utf-8 -*-
2 from __future__ import absolute_import, division, print_function
10 sys.path.append('../../')
12 from libs.configs import cfgs
16 if not os.path.exists(path):
22 image_path = os.path.join(cfgs.ROOT_PATH, '{}/JPEGImages'.format(cfgs.DATASET_NAME))
23 xml_path = os.path.join(cfgs.ROOT_PATH, '{}/Annotations'.format(cfgs.DATASET_NAME))
25 image_list = os.listdir(image_path)
27 image_name = [n.split('.')[0] for n in image_list]
29 random.shuffle(image_name)
31 train_image = image_name[:int(math.ceil(len(image_name)) * divide_rate)]
32 test_image = image_name[int(math.ceil(len(image_name)) * divide_rate):]
34 image_output_train = os.path.join(
35 cfgs.ROOT_PATH, '{}_train/JPEGImages'.format(cfgs.DATASET_NAME))
36 mkdir(image_output_train)
37 image_output_test = os.path.join(
38 cfgs.ROOT_PATH, '{}_test/JPEGImages'.format(cfgs.DATASET_NAME))
39 mkdir(image_output_test)
41 xml_train = os.path.join(cfgs.ROOT_PATH, '{}_train/Annotations'.format(cfgs.DATASET_NAME))
43 xml_test = os.path.join(cfgs.ROOT_PATH, '{}_test/Annotations'.format(cfgs.DATASET_NAME))
49 shutil.copy(os.path.join(image_path, i + '.jpg'), image_output_train)
50 if os.path.exists(os.path.join(xml_path, i + '.xml')):
51 shutil.copy(os.path.join(xml_path, i + '.xml'), xml_train)
53 print("process step {}".format(count))
57 shutil.copy(os.path.join(image_path, i + '.jpg'), image_output_test)
58 shutil.copy(os.path.join(xml_path, i + '.xml'), xml_test)
60 print("process step {}".format(count))