背景:视频文件处理成一帧一帧的图像做训练,图片文件文件名是医xxxx_1.jpg~xxxx_999.jpg,划分方式如下,图片文件名以2和9结尾的文件划为val数据,图片文件以4,8结尾划入test数据,代码如下
#将特定文件夹下的文件移动到另一个文件夹
import os
import shutil
path1 = "XXXXXX"
filelist = os.listdir(path1)
path2 = "XXXX"
for files in filelist:
filename1 = os.path.splitext(files)[1]XXXXXX # 读取文件后缀名
filename0 = os.path.splitext(files)[0] #读取文件名
print(filename1)
m = filename0[-1]
print(m)
if m == '4' or m == '8':
full_path = os.path.join(path1, files)
print(full_path)
despath = path2 + files
print(despath)
shutil.move(full_path, despath)
else :
continue