程序笔记   发布时间:2022-07-19  发布网站:大佬教程  code.js-code.com
大佬教程收集整理的这篇文章主要介绍了python进行文件分类大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。

python将数据快速分成训练集和测试集

博客: python将一个文件夹下图片按比例分在两个文件夹下_qq_32039433的博客-CSDN博客

代码如下:

神经网络模型训练时得到的数据并没有分成测试集和训练集,所以需要自己按比例进行分割,如下是进行的按照3:7进行数据分割python代码。

#!/usr/bin/env python# -*- coding:utf-8 -*-# 将一个文件夹下图片按比例分在两个文件夹下,比例改0.7这个值即可import osimport randomimport shutilfrom shutil import copy2Trainfiles = os.listdir('C:/Users/cozce/Desktop/机器学习/第二次 实验大作业-船舶分类(CNN)/船舶数据(船和非船类)/sea')#(图片文件夹)num_Train = len(Trainfiles)print( "num_Train: " + str(num_Train) )index_list = list(range(num_Train))print(index_list)random.shuffle(index_list)num = 0TrainDir = 'C:/Users/cozce/Desktop/机器学习/第二次 实验大作业-船舶分类(CNN)/船舶数据(船和非船类)/sea_Train'#(将图片文件夹中的7份放在这个文件夹下)validDir = 'C:/Users/cozce/Desktop/机器学习/第二次 实验大作业-船舶分类(CNN)/船舶数据(船和非船类)/sea_test'#(将图片文件夹中的3份放在这个文件夹下)for i in index_list:    filename = os.path.join('C:/Users/cozce/Desktop/机器学习/第二次 实验大作业-船舶分类(CNN)/船舶数据(船和非船类)/sea', Trainfiles[i])    if num < num_Train*0.7:        print(str(fileName))        copy2(filename, TrainDir)    else:        copy2(filename, validDir)    num += 1

另附上批量修改图片名代码:

import osimport reimport syspath = r"D:Code机器学习VGG16VGG16_1Is_shipTrainsea"fileList = os.listdir(path)  # 待修改文件夹print("修改前:" + str(fileList))  # 输出文件夹中包含的文件os.chdir(path)  # 将当前工作目录修改为待修改文件夹的位置num = 1  # 名称变量for filename in fileList:  # 遍历文件夹中所有文件    pat = ".+.(jpg|PNG|png)"  # 匹配文件名正则表达式    pattern = re.findall(pat, fileName)  # 进行匹配    print('pattern[0]:', pattern)    print('num:', num, 'filename:', fileName)    os.rename(filename, ('sea.' + str(num)+".png"))  # 文件重新命名    num = num + 1  # 改变编号,继续下一项print("---------------------------------------------------")sys.stdin.flush()  # 刷新print("修改后:" + str(os.listdir(path)))  # 输出修改后文件夹中包含的文件

 

大佬总结

以上是大佬教程为你收集整理的python进行文件分类全部内容,希望文章能够帮你解决python进行文件分类所遇到的程序开发问题。

如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。