python获取图片大小&放到另外一个文件夹里面

最近在做分类训练时,由于人脸检测算法检测和截图出来的图片有一些比较小,不利于分类训练,所以我将36像素以上的像素的图片单独给放到一个文件夹里,总共图片有3w+张,所以写个python来筛选吧。
用caffe搞训练的我觉得看我的代码就好啦,如果不是,看参考资料的代码吧。

【0、参考资料】

获取文件大小
复制图片到另外一个文件夹

【1、安装依赖库】

pip install pillow

【2、代码】

import os
from PIL import Image
import shutil
file_path = '/home/your/file/path/' 
target_path = '/home/your/target/path/' #放筛选后的图片的文件夹
count = 0
with open(file_path + 'test_001_30000.txt', 'r') as f: #这个文件里,每一行是图片的一个名字和label
    for line in f:
        img_path = line.split(' ')[0]
        img = Image.open(file_path + 'test/' + img_path)
        size = img.size #获取图片大小,是一个元组
        temp = min(size)
        if temp >= 35:
            shutil.copy(file_path + 'test/' + img_path, target_path) #复制到 另外一个文件夹
            count += 1
        if count != 0 and count % 32 == 0:
            print('count = {}'.format(count))

https://www.jianshu.com/p/db14a84426e8

Python量化投资网携手4326手游为资深游戏玩家推荐:《《梦间集》星罗棋布版本更新说明

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论