将图片切割

最新推荐文章于 2022-11-29 21:20:34 发布

Fairy爱学习

最新推荐文章于 2022-11-29 21:20:34 发布

阅读量121

点赞数

分类专栏：数据处理文章标签： python

本文链接：https://blog.csdn.net/weixin_44118897/article/details/119532929

版权

数据处理专栏收录该内容

5 篇文章 0 订阅

订阅专栏

import cv2
import os

if __name__ == "__main__":
    path = r"C:\Users\ZQ\Desktop\RSDDs dataset\Type-I RSDDs dataset\luo\img"  # change this dirpath.
    listdir = os.listdir(path)

    newdir = os.path.join(path, 'split1_groundtruth')  # make a new dir in dirpath.
    if (os.path.exists(newdir) == False):  # 如果文件不存在，则创建文件。
        os.mkdir(newdir)

    for i in listdir:
        if i.split('.')[1] == "png" or i.split('.')[1] == "JPG" or i.split('.')[1] == "jpg" :
            filepath = os.path.join(path, i)
            filename = i.split('.')[0]
            leftpath = os.path.join(newdir, filename) + "_up.png"
            rightpath = os.path.join(newdir, filename) + "_down.png"

            img = cv2.imread(filepath)
            [h, w] = img.shape[:2]
            print(filepath, (h, w))
            # limg = img[:int(h / 2), :int(w / 2), :] # 左上
            # rimg = img[int(h / 2 + 1):, :int(w / 2), :] # 左下
            # limg = img[:int(h / 2), int(w / 2 + 1):, :]  # 右上
            # rimg = img[int(h / 2 + 1):, int(w / 2 + 1):, :]  # 右下
            limg = img[:int(h / 2), :, :]  # 上
            rimg = img[int(h / 2 + 1):, :, :]  # 下

            imagel = cv2.resize(limg, (840, 840), interpolation=cv2.INTER_CUBIC)
            imager = cv2.resize(rimg, (840, 840), interpolation=cv2.INTER_CUBIC)

            cv2.imwrite(leftpath, imagel)
            cv2.imwrite(rightpath, imager)