python程序代码图片_python实现图片筛选程序

#_*_coding:utf-8_*_

'''

Version V17.1.0

Date 2017-10-15

@Author Cheney

'''

# 收集EL 图片集中需要的图片，在elpick目录下进行

from PIL import Image

import csv

import os

# 获取需要图片的文件名

def readImageName():

# 在elpick目录下读取sn.scv文件，该文件读取出来后元素是list 需要转换为str

get_csv = csv.reader(open("elpick/sn.csv"))

origSN = [] transSN = [] for sn in get_csv:

origSN.append(sn)

tempList = origSN

# 读取csv文件获时获取到的列表，其元素也是列表，需要将其转换为字符串元素，使用下面for语句实现

for s in tempList:

# 去掉列表元素中的[''] 这几个字符

s = str(s).strip('['')

s = str(s).strip('']')

transSN.append(s)

ImageName = transSN

return ImageName

pickImageName = readImageName()

# 遍历所有图片集的文件名

for image in os.listdir("elpick/original_el"):

# 将读取到的文件名进行处理，去扩展名，若是int文件名需要对文件名进行int 处理

imageName = image.strip(".jpg")

# 比较判断 若遍历的文件名在需要收集的列表中，读取之后并保存到pick_el文件夹下

if imageName in pickImageName:

pickImage = Image.open(('elpick/original_el/%s.jpg') % imageName)

pickImage.save(("elpick/picked_el/%s.jpg") % imageName)