python实现根据坐标将图片进行裁剪

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])
    # 进行排序
    # key=lambda box: (int(box[1]), int(box[0])) 表示：
    #   1. 先按 box 的第 1 个元素（y1）进行升序排序（从上到下）
    #   2. 如果 y1 相同，再按第 0 个元素（x1）进行升序排序（从左到右）
    #   3. int() 是为了将字符串转为数字，确保正确排序
    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()
    # 遍历排序后的列表，进行编号（从1开始）
    # 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"编号: {index}, 坐标: {box}")



    # --- 主要修改和新增的逻辑在这里 ---
    for index, box in enumerate(sorted_boxes, 1):
        # 1. 将字符串坐标转换为整数
        x1, y1, x2, y2 = map(int, box)

        # 2. 【关键修复】确保坐标顺序正确，防止计算负数半径
        # 重新计算左上角和右下角坐标，确保 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. 绘制矩形框（使用修正后的坐标）
        cv2.rectangle(copy_img, (top_left_x, top_left_y), (bottom_right_x, bottom_right_y), (0, 0, 255), 2)

        # --- 新增代码开始 ---
        # 4. 计算矩形中心点
        center_x = (top_left_x + bottom_right_x) // 2
        center_y = (top_left_y + bottom_right_y) // 2

        # 7. 在圆圈中心写入编号
        font = cv2.FONT_HERSHEY_SIMPLEX
        text = str(index)
        # 计算文字大小，以便居中
        (text_width, text_height), baseline = cv2.getTextSize(text, font, 1, 2)

        # 计算文字的坐标，使其在圆圈内居中
        text_x = center_x - text_width // 2
        text_y = center_y + text_height // 2

        # 绘制文字
        # 参数：图像, 文字, 坐标, 字体, 大小, 颜色(白色), 粗细
        cv2.putText(copy_img, text, (text_x, text_y), font, 10, (0, 0, 255), 3)
        # --- 新增代码结束 ---

        print(f"编号: {index}, 原始坐标: {box}, 修正后坐标: {[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()