如何在 Python 中裁剪和存储边界框图像区域？

Question

我的想法是使用给定图像的异常区域的多个边界框坐标并将这些区域裁剪以保存到单独的文件夹中。 我编写了如下所示的代码，为单个图像裁剪这些多个边界框坐标，但是，我也得到了我必须摆脱的边界框。

import pandas as pd
import cv2
import numpy as np

df = pd.read_csv('excel1.csv')
image = cv2.imread('image2.png')
im_name = 'image2.png'

for i in range(len(df)):
    name = df.loc[i]['filename']
    if name == im_name:
        start_point = (df.loc[i]['x'],df.loc[i]['y'])  
        end_point = (df.loc[i]['x']+df.loc[i]['width'],df.loc[i]['y']+df.loc[i]['height'])  
        color = (128, 0, 0)  
        thickness = 2  
        image = cv2.rectangle(image, start_point, end_point, color, thickness) 
        crop = image[df.loc[i]['y']:df.loc[i]['y']+df.loc[i]['height'],
                     df.loc[i]['x']:df.loc[i]['x']+df.loc[i]['width']]
        cv2.imwrite("cropped/crop_{0}.png".format(i), crop)  
cv2.imwrite('bb.png', image)

Answer 1

在循环中使用 numpy 切片，然后使用 Python/OpenCV imwrite() 在循环内使用不同的名称对循环的每次迭代进行裁剪

crop = image[ystart:ystop, xstart:xstop]
cv2.imwrite("crop_{0}.png".format(i), crop)

如果您希望它们 go 到不同的文件夹，您还可以为要写入的每个图像添加不同的路径。

numpy切片见https://www.w3schools.com/python/numpy_array_slicing.asp

Answer 2

我找到了解决办法。 我删除 cv2.rectangle 只是因为我想只存储边界框区域，这样边界框就不会出现。