Python iter_direct_files示例

示例#1

def main():
    for _, p in iter_direct_files(DATA_PREPROCESSED):
        if _ in handled:
            continue
        im = Image.open(p)
        im.show()
        try:
            code = raw_input('验证码：')
        except KeyboardInterrupt:
            with open(handled_tmp, 'w') as f:
                pickle.dump(handled, f)
            break
        else:
            im.save(os.path.join(DATA_NAMED, code), 'jpeg')
            handled.add(_)
            print 'code:', code

示例#2

文件： preprocess.py 项目： Zephor5/seccode_recognize

# coding=utf-8
import os

from PIL import Image, ImageEnhance

from conf import DATA_RAW, DATA_PREPROCESSED
from tools import iter_direct_files

__author__ = 'zephor'


def process(img, img_path):
    im = Image.open(img_path)
    sharp_enhancer = ImageEnhance.Sharpness(im)
    im = sharp_enhancer.enhance(2)
    im = im.resize((200, 46), Image.LANCZOS)
    im.save(os.path.join(DATA_PREPROCESSED, img))


if __name__ == '__main__':
    for fn, p in iter_direct_files(DATA_RAW):
        process(fn, p)