ArcPy脚本根据影像文件路径自动创建金字塔

代码

# -*-coding:gbk-*-

import arcpy
import os
import sys
import time

image_path = sys.argv[1]

# image_path = u"E:/Temp/20190415GD三调质检/440103LWQ/440103KM4+GF2+GE1DOM01.IMG"
# image_path = u"E:/Temp/20190415GD三调质检/440607SSQ高清影像"

if __name__ == '__main__':
    try:
        time_begin = time.time()
        if not os.path.exists(image_path):
            raise Exception(u"Path {0} not exist".format(image_path))
        files = []
        if os.path.isfile(image_path):
            files.append(image_path)
        elif os.path.isdir(image_path):
            for dir_path, dir_names, file_names in os.walk(image_path):
                file_names = filter(lambda file_name: file_name[-4:].upper() in [".IMG", ".TIF"], file_names)
                file_names = map(lambda file_name: os.path.join(dir_path, file_name), file_names)
                files.extend(file_names)
        print  u"{0}".format(files)
        if len(files) > 0:
            for file_path in files:
                arcpy.BuildPyramids_management(file_path, 16, None, None, None, None, "SKIP_EXISTING")
        print "Total Time {0}".format(time.time() - time_begin)
        print "1"
    except Exception, e:
        print  u"{0}".format(e.message)

说明

• 传入的参数可为文件夹或文件。
• 支持IMG、TIF两种格式。
• 默认创建影像级别为16级。
• 已创建金字塔的跳过。