opencv - 图片中的Scipy绘制边框

Question

我之前使用cv.rectangle() OpenCV方法在numpy数组中绘制边框，然后将其保存到文件中。但是，我已经开始用scipy替换OpenCV操作，并且在scipy中无法轻松找到与此等效的方法。有没有办法可以在scipy中实现？

Answer 1

您可以通过使用简单的矩阵操作操作并用给定的颜色替换所需的行和列来完成此操作:

from scipy.misc import imsave
import numpy as np

# Create 500 x 500 Empty canvas of white color
arr = np.ones((500, 500, 3), dtype=np.uint8) * 255
color = np.array([0, 255, 0], dtype=np.uint8)
bounding_box = (100, 100, 200, 200)

arr[bounding_box[1], bounding_box[0]:bounding_box[0] + bounding_box[2]] = color
arr[bounding_box[1]:bounding_box[1] + bounding_box[3], bounding_box[0]] = color

arr[bounding_box[1] + bounding_box[3], bounding_box[0]:bounding_box[0] + bounding_box[2]] = color
arr[bounding_box[1]:bounding_box[1] + bounding_box[3], bounding_box[0] + bounding_box[2]] = color

imsave("./debug.png", arr)

输出: