按官方给的TrainImages.txt与TestImages.txt来划分,图像数据集Images文件夹,以下程序与这些文件放在同级位置,划分后得到训练集5360张,测试集1340张,在Windows下执行的python程序:
import os
import shutil
path = 'D:\\MIT67'
data_dir = path + '\\Images'
train_dir = path + '\\train'
test_dir = path + '\\test'
train_txt_path = path + '\\TrainImages.txt'
test_txt_path = path + '\\TestImages.txt'
if not os.path.exists(train_dir):
os.makedirs(train_dir)
if not os.path.exists