数据集划分,文件移动

背景:视频文件处理成一帧一帧的图像做训练,图片文件文件名是医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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值