import os import shutil img_dir = "C:/Users/lirui/desktop/data/train/imgs" mask_dir = "C:/Users/lirui/desktop/data/train/masks" test_img_dir = "C:/Users/lirui/desktop/data/test/imgs" test_mask_dir = "C:/Users/lirui/desktop/data/test/masks" img_list = sorted(os.listdir(img_dir)) test_ratio = 4 for idx,img_name in enumerate(img_list): if idx % test_ratio == 0: shutil.move(os.path.join(img_dir,img_name),os.path.join(test_img_dir,img_name)) shutil.move(os.path.join(mask_dir,img_name),os.path.join(test_mask_dir,img_name))
训练测试划分
最新推荐文章于 2024-07-21 22:27:18 发布