首页 > 代码库 > python如何讲一个文件中的图片分到两个
python如何讲一个文件中的图片分到两个
最近在做一个图像分类的比赛,作为初次接触深度学习的菜鸟,上手了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‘目录下。
python如何讲一个文件中的图片分到两个
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。