如何在 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)

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

对于 numpy 切片，请参阅 https://www.w3schools.com/python/numpy_array_slicing.asp

Answer 2

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

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

How to crop and store bounding box image regions in Python?

python

opencv

crop

bounding-box