python读取大图按照txt文件保存的location截图并保存

一、文件结构

二、location.txt

按照（x1,x2,y1,y2,name）

1633,2067,1923,3016,a
2846,3666,3391,4439,b
3197,2128,3786,2550,c

三、screenShot.py

import cv2


def imgRead(img_path):
    """读取图像"""
    imgCV = cv2.imread(img_path)
    return imgCV

def lineRead(location_path):
    """读取位置文件,location: (x1,y1,x2,y2)"""
    locs = []
    for line in open(location_path):
        locs.append(line.strip())

    return locs

def screenShot(img,locs,savePath):
    """截图并保存图片"""
    for locStr in locs:
        img_loc = locStr.split(',') # 字符串列表
        print(img_loc)
        ymin, ymax ,xmin, xmax = int(img_loc[1]),int(img_loc[3]),int(img_loc[0]),int(img_loc[2])

        img_crop = img[ymin:ymax,xmin:xmax]
        cv2.imwrite(savePath +img_loc[4]+'.bmp',img_crop)


if __name__ == '__main__':
    """读取图像"""
    img_path = './Image__2021-05-06.bmp'
    img = imgRead(img_path)


    """读取位置信息"""
    loc_path = './location.txt'
    locs = lineRead(loc_path)

    """根据位置信息分割图像并存储"""
    import os
    save_path = 'IMG1/'
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    screenShot(img,locs,save_path)