1.根据路径获得所有图像的列表
2.根据比例去获得训练集和验证集的列表(切片)
3.根据新划分的列表把图片移到相应的文件夹下
用复制不用移动,可以只对样本划分训练集和验证集,标签文件是总的,最后通过遍历训练集或验证集找其对应的标签。
import os
import shutil
列表生成式:
[expression for item in iterable if condition]
os.listdir("目录名") 获得该目录下所有文件名字的列表,可以利用列表生成式形成完整相对路径的列表
os.makedirs("arr/brr/vrr") 生成多级文件目录结构
open("arr/brr/am.txt",'w') 生成文件
shutil.move("arr" , "brr/") 可以把文件或目录转移到另一目录下,记得后面加/
shutil.copy("arr.py","brr/") 可以把文件复制到另一目录下,记得后面加/
shutil.copytree("arr","brr/arr") 可以把整个文件结构复制到另一目录下,记得后面加上/"需要复制的目录名"
os.rename("arr","brr") 可以更改文件或文件夹的名字
os.remove("arr.txt") 删除文件
shutil.rmtree("arr") 删除整个文件结构