python 区分图像大小(A2、A3、

发布时间:2019-09-18 07:21:09编辑:auto阅读(1852)

    说明,本人对象负责的项目有大量的加工图像,分别有A2 A3 A4 等规格,且这些图像都是在一起存储,按照相关的档案顺序全组;现在让我分别统计一共的图像数量 以及A2  A3  A4数量,经过一晚努力,现将代码公布如下:

    这里主要用到了Image这个模块,需要自行下载

    路径注意事项:在Windows里,需要将路径中的\全部改成//

    适用场景:有大量图像,需要区分出A3 A4 对应的数量
    实现手段:遍历目录,并打开文件比较其像素大小值
    功能的实现:除了按照像素值比较图像大小区分外,
    额外功能的增加:1加入了只针对.jpg图像进行处理,即使用其它格式的文件也不会影响程序执行;
                                2加入了异常处理,即文件是图像,但是无法打开,这块会定义它为错误图像;这里利用了try,即程序不会因为图像打不开而中断
    实现语言:python

    样例:
    #!/usr/bin/env python


    import os
    import os.path
    import Image
    A1=0
    A2=0
    A3=0
    A4=0
    total = 0
    errornum = 0

    rootdir = "F://数据备份"
    errorfile = 'd://图像错误.txt'

    for parent,dirnames,filenames in os.walk(rootdir):
        if len(filenames) > 0 and len(dirnames) == 0:
            for i in range(len(filenames)):
                filename = parent + "//" + filenames[i]
                if filename.find(".jpg") > 0:
                    total+=1
                    print filename
                    try:
                        p_w_picpath = Image.open(filename)             
                    except :
                        errornum = errornum + 1
                        error = file(errorfile,'a')
                        error.write('\r\n')
                        error.write(filename)
                        error.flush
                        continue
                    p_w_picpath_size = p_w_picpath.size[0]
                    if p_w_picpath_size <= 3500:
                         A4 = A4 + 1
                    elif p_w_picpath_size > 3500 and p_w_picpath_size <= 6000:
                         A3 = A3 + 1
                    elif p_w_picpath_size > 6000 and p_w_picpath_size <= 9000:
                         A2 = A2 + 1
                    elif p_w_picpath_size > 9000:
                         A1 = A1 + 1
                else:
                    continue
    print "-------------------------------------------------"
    print "总数如下:%d\r"% total
    print "A1一共有:%d\r"% A1
    print "A2一共有:%d\r"% A2
    print "A3一共有:%d\r"% A3
    print "A4一共有:%d\r"% A4
    print '错误图像有共有:%d\r'% errornum

关键字