python如何讲一个文件中的图片分到两个
2024-08-27 23:52:33
最近在做一个图像分类的比赛,作为初次接触深度学习的菜鸟,上手了keras。说实话,除了keras教程,中文博客的技术支持太差了。正在头大的学习中...废话不多说,记录一下学习中的一些小细节。在遇到generator.flow_from_directory('data/train'...)这个函数时,需要输入训练图片的路径。此时训练图像和验证图像都在一个目录下,所以我需要重建两个文件来分别放训练图像和验证图像。开始,我想,利用opencv将每张图片读出来,然后再根据是否为训练图像或者验证图像分别重新写到重新建的文件夹里。后来发现python中的 from shutil import copy2 函数可以直接将图片放到指定的路径下。
import os
import random
import shutil
from shutil import copy2
trainfiles = os.listdir('Data/train')
num_train = len(trainfiles)
index_list = range(num_train)
random.shuffle(index_list)
num = 0
trainDir = 'Data/pre_train'
validDir = 'Data/pre_valid'
for i in index_list:
fileName = os.path.join('Data/train', trainfiles[i])
if num < num_train*0.8:
copy2(fileName, trainDir)
else:
copy2(fileName, validDir)
num += 1
简单思路:将原来‘Data/train’路径下的图片名读出,然后打乱所有图片的顺序,将打乱顺序后的总图片数的前80%图片作为训练数据放到新建的'Data/pre_train'目录下,其余的图片放到'Data/pre_valid'目录下。
最新文章
- 新年抢红包效果(New Year Red Packet)
- 转载:Hadoop排序工具用法小结
- (原) 2.3 Curator使用
- error CS0117: `UnityEditor.EditorUtility&#39; does not contain a definition for `GetAssetPreview&#39;
- js-innerHTML
- BZOJ-2037 Sue的小球 DP+费用提前
- poj3207 2-SAT入门
- python 根据对象和方法名,返回提供这个方法的定义的类
- jsp的el表达式
- Axure自动幻灯片制作
- poi读写Excel文件
- Office文档在线编辑的实现之一
- 【洛谷1131】【ZJOI2007】时态同步
- Sitecore&#174; 8.2 Professional Developer考试心得
- c++11の泛型算法
- Multi-Get API
- Paper | 多任务学习的鼻祖
- &#39;新&#39;, &#39;泽&#39; - ImageMagick - UTF-8非最短形式及编码安全问题
- jdk8- list操作
- Leetcode——121. 买卖股票的最佳时机
热门文章
- oracle游标的使用
- v-for并判断当前元素是否选中:$set实现响应添加属性
- 解决nginx: [error] open() "/usr/local/nginx/logs/nginx.pid" failed错误
- 将自己的代码托管到github上
- asp.net -mvc框架复习(6)-基于MVC实现简单计算器
- 【编程技巧】Ext.MessageBox 大集合 不同的dialog图解加写法
- maven多模块搭建
- linux下ftp命令的安装与使用
- WebRoot 与 webContent的区别
- LAMP_yum安装