cv2.rectangle()不会执行任何操作

Question

我有一个python脚本，它看起来像这样：

import cv2
if __name__ == "__main__":
predict_filename = "prediction.jpg"
predict_cv2 = cv2.imread(prediction_filename)

cv2.rectangle(predict_cv2, (50,50), (100,100), (225,0,225), 6)
print("changing ", predict_filename , " done")

我稍后会尝试绘制方框来显示对象检测结果，但直到现在，我还很难在图片上绘制方框。prediction.jpg与脚本存在于同一个文件中，但在运行代码时将保持不变。我的错误在哪里？

Answer 1

您正在predict_cv2的numpy数组上绘制矩形，如果您想保存此图像，则需要使用命令cv2.imwrite("文件名"，要保存的数组(

import cv2
if __name__ == "__main__":
predict_filename = "prediction.jpg"
predict_cv2 = cv2.imread(prediction_filename)

cv2.rectangle(predict_cv2, (50,50), (100,100), (225,0,225), 6)
cv2.imwrite("ImageWithRectangle.jpg", predict_cv2)
print("changing ", predict_filename , " done")

这应该会将带有矩形的图像保存在项目文件夹中。

Answer 2

您所需要做的就是在draw矩形命令之前使用cv2.imwrite语句来获得所需的结果。

cv2.imwrite("File_name.jpg", predict_cv2)