| import random |
| import shutil |
| import os |
|
|
| def split(path, mask_path, lane_path): |
| os.mkdir(path + 'train') |
| os.mkdir(path + 'val') |
| os.mkdir(mask_path + 'train') |
| os.mkdir(mask_path + 'val') |
| os.mkdir(lane_path + 'train') |
| os.mkdir(lane_path + 'val') |
| val_index = random.sample(range(660), 200) |
| for i in range(660): |
| if i in val_index: |
| shutil.move(path+'{}.png'.format(i), path + 'val') |
| shutil.move(mask_path+'{}.png'.format(i), mask_path + 'val') |
| shutil.move(lane_path+'{}.png'.format(i), lane_path + 'val') |
| else: |
| shutil.move(path+'{}.png'.format(i), path + 'train') |
| shutil.move(mask_path+'{}.png'.format(i), mask_path + 'train') |
| shutil.move(lane_path+'{}.png'.format(i), lane_path + 'train') |
|
|
|
|
| if __name__ == '__main__': |
| path = "/home/wqm/bdd/data_hust/" |
| mask_path = "/home/wqm/bdd/hust_area/" |
| lane_path = "/home/wqm/bdd/hust_lane/" |
| split(path, mask_path, lane_path) |
|
|
|
|
|
|
