裁剪图片

# - *-coding=utf-8 -*-
'''
将瑕疵图片的boundingbox截取下来作为新的一张图片
同时resize到跟原始图片统一大小
'''

import cv2
import os

def show_pic(img, bboxes=None):
    '''
    输入:
        img:图像array
        bboxes:图像的所有boudning box list, 格式为[[x_min, y_min, x_max, y_max]....]
        names:每个box对应的名称
    '''
    cv2.imwrite('./1.jpg', img)
    img = cv2.imread('./1.jpg')
    for i in range(len(bboxes)):
        bbox = bboxes[i]
        x_min = bbox[0]
        y_min = bbox[1]
        x_max = bbox[2]
        y_max = bbox[3]
        cv2.rectangle(img,(int(x_min),int(y_min)),(int(x_max),int(y_max)),(0,255,0),3)
    cv2.namedWindow('pic', 0)  # 1表示原图
    cv2.moveWindow('pic', 0, 0)
    cv2.resizeWindow('pic', 1200,800)  # 可视化的图片大小
    cv2.imshow('pic', img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    os.remove('./1.jpg')

def crop_bd(img, bboxes):
        '''
        裁剪后的图片为boudingbox围起来的区域
        输入:
            img:图像array
            bboxes:该图像包含的所有boundingboxs,一个list,每个元素为[x_min, y_min, x_max, y_max],要确保是数值
        输出:
            crop_bds:裁剪后的boudingbox图像array list,（可能有多个boudingbox）
        '''
        #---------------------- 裁剪图像 ----------------------
        crop_bds = list()
        for bbox in bboxes:
            x_min = bbox[0]
            y_min = bbox[1]
            x_max = bbox[2]
            y_max = bbox[3]
            crop_bd = img[y_min:y_max, x_min:x_max]
            crop_bds.append(crop_bd)
        return crop_bds

if __name__ == '__main__':

    from xml_helper import *
    import shutil

    source_bad_pic_root_path = 'F:/mx_matting/yolov3-master/data/JPEGImages'
    source_xml_root_path = 'F:/mx_matting/yolov3-master/data/Annotations'

    target_pic_root_path = 'F:/mx_matting/yolov3-master/data//data_croped'  #只包括有瑕疵的图片
    if os.path.exists(target_pic_root_path):
        shutil.rmtree(target_pic_root_path)
    os.mkdir(target_pic_root_path)

    cnt = 0
    process_flag = 0
    for parent, _, files in os.walk(source_bad_pic_root_path):
        for file in files:
            process_flag += 1
            print(str(process_flag)+'/'+str(len(files)))
            bad_pic_path = os.path.join(parent, file)
            xml_path = os.path.join(source_xml_root_path, file[:-4]+'.xml')
            img = cv2.imread(bad_pic_path)
            bboxes = parse_xml(xml_path)

            # #原图可视化一下
            # show_pic(img,bboxes)

            croped_imgs = crop_bd(img, bboxes)
            for croped_img in croped_imgs:
                cnt += 1
                target_pic_path = os.path.join(target_pic_root_path, file[:-4]+'_croped'+str(cnt)+'.jpg')
                # #reize一下
                # croped_img = cv2.resize(croped_img, (img.shape[1], img.shape[0]))
                # #可视化一下截取的图
                # show_pic(croped_img,[[0,0,croped_img.shape[1],img.shape[0]]])
                #写入
                cv2.imwrite(target_pic_path, croped_img)