python實(shí)現(xiàn)根據(jù)坐標(biāo)將圖片進(jìn)行裁剪

更新時(shí)間：2025年11月10日 09:18:55 作者：NO1212

這篇文章主要為大家詳細(xì)介紹了如何使用python實(shí)現(xiàn)根據(jù)坐標(biāo)將圖片進(jìn)行裁剪,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

功能：

1.給一個(gè)圖片，并給出圖片中所有目標(biāo)的坐標(biāo)

2.將圖中給出的坐標(biāo)用框給標(biāo)注出來(lái)

3.經(jīng)典需求，通常稱為“閱讀順序排序”或“Z字排序”。你需要先按行（從上到下）排序，在同一行內(nèi)再按列（從左到右）排序。將排完序的多個(gè)框進(jìn)行編號(hào)

4.對(duì)目標(biāo)框進(jìn)行裁圖

完整代碼：

import cv2
import numpy as np
import os


def test():
    path = "D:\LAELAOI\Test"
    pointtxt = os.path.join(path, "label_position.txt")
    with open(pointtxt, 'r')as f:
        content = f.readlines()
    print(content)
    boxes = []
    for i in content:
        parts = i.split(',')
        file = parts[0]
        x1, x2, y1, y2 = parts[2],parts[3],parts[4],parts[5].strip()
        print(file, x1, y1, x2, y2)
        boxes.append([x1, y1, x2, y2])
    # 進(jìn)行排序
    # key=lambda box: (int(box[1]), int(box[0])) 表示：
    #   1. 先按 box 的第 1 個(gè)元素（y1）進(jìn)行升序排序（從上到下）
    #   2. 如果 y1 相同，再按第 0 個(gè)元素（x1）進(jìn)行升序排序（從左到右）
    #   3. int() 是為了將字符串轉(zhuǎn)為數(shù)字，確保正確排序
    sorted_boxes = sorted(boxes, key=lambda box: (int(box[1]), int(box[0])))
    img = cv2.imread(r"D:\LAELAOI\Test\L_main_CAM_basler.jpg")
    # img = cv2.resize(img, (1500, 1000))
    copy_img = img.copy()
    # 遍歷排序后的列表，進(jìn)行編號(hào)（從1開(kāi)始）
    # for index, box in enumerate(sorted_boxes, 1):
    #     x1, x2, y1, y2 = int(box[0]), int(box[1]), int(box[2]), int(box[3])
    #     cv2.rectangle(copy_img,(int(box[0]),int(box[1])), (int(box[2]),int(box[3])), (0,0, 255), 2)
    #     print(f"編號(hào): {index}, 坐標(biāo): {box}")



    # --- 主要修改和新增的邏輯在這里 ---
    for index, box in enumerate(sorted_boxes, 1):
        # 1. 將字符串坐標(biāo)轉(zhuǎn)換為整數(shù)
        x1, y1, x2, y2 = map(int, box)

        # 2. 【關(guān)鍵修復(fù)】確保坐標(biāo)順序正確，防止計(jì)算負(fù)數(shù)半徑
        # 重新計(jì)算左上角和右下角坐標(biāo)，確保 x1 <= x2, y1 <= y2
        top_left_x = min(x1, x2)
        top_left_y = min(y1, y2)
        bottom_right_x = max(x1, x2)
        bottom_right_y = max(y1, y2)

        # 3. 繪制矩形框（使用修正后的坐標(biāo)）
        cv2.rectangle(copy_img, (top_left_x, top_left_y), (bottom_right_x, bottom_right_y), (0, 0, 255), 2)

        # --- 新增代碼開(kāi)始 ---
        # 4. 計(jì)算矩形中心點(diǎn)
        center_x = (top_left_x + bottom_right_x) // 2
        center_y = (top_left_y + bottom_right_y) // 2

        # 7. 在圓圈中心寫(xiě)入編號(hào)
        font = cv2.FONT_HERSHEY_SIMPLEX
        text = str(index)
        # 計(jì)算文字大小，以便居中
        (text_width, text_height), baseline = cv2.getTextSize(text, font, 1, 2)

        # 計(jì)算文字的坐標(biāo)，使其在圓圈內(nèi)居中
        text_x = center_x - text_width // 2
        text_y = center_y + text_height // 2

        # 繪制文字
        # 參數(shù)：圖像, 文字, 坐標(biāo), 字體, 大小, 顏色(白色), 粗細(xì)
        cv2.putText(copy_img, text, (text_x, text_y), font, 10, (0, 0, 255), 3)
        # --- 新增代碼結(jié)束 ---

        print(f"編號(hào): {index}, 原始坐標(biāo): {box}, 修正后坐標(biāo): {[top_left_x, top_left_y, bottom_right_x, bottom_right_y]}")
        cropped_img = img[y1:y2, x1:x2]
        savefile = r"D:\LAELAOI\Test\{}.jpg".format(index)
        print(savefile)
        cv2.imwrite(savefile, cropped_img)
    cv2.imwrite(r"D:\LAELAOI\Test\1111L_main_CAM_basler.jpg", copy_img)
    cv2.imshow('rr', copy_img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
if __name__ == '__main__':
    test()

效果圖：