如何正确使用cv2.imwrite和cv2.selectROI在openCV中保存图像

Question

我正在尝试使用OpenCV的ROI函数。使用它，我试图裁剪加载的图像的一部分。之后，我试图保存图像并显示它。显示它不是太大的问题，但保存它却是一个问题。该图像被存储为一个大黑矩形，而不是实际裁剪后的图像。以下是我的代码：

import cv2
import numpy as np
from skimage.transform import rescale, resize

if __name__ == '__main__' :

    # Read image
    im = cv2.imread("/Path/to/Image.jpg")
    img = resize(im, (400,400), mode='reflect') 
    # Select ROI
    r = cv2.selectROI(img)

    # Crop image
    imCrop = img[int(r[1]):int(r[1]+r[3]), int(r[0]):int(r[0]+r[2])]

    # Save first, then Display cropped image
    cv2.imwrite("../../Desktop/Image.jpg", imCrop) # This is where there seems to be a problem
    cv2.imshow("im", imCrop)
    cv2.waitKey(0)

有人可以帮忙吗？

- TheTank

---

你能检查一下 imCrop 的内容吗？它是整数还是浮点数（在0到1之间）。另外，为了确保万无一失，你可以尝试将写入格式指定为jpeg吗？ - Umang Gupta

它原本是介于0到1之间的浮点数，我将其转换为了0到255。 - TheTank

那样做能让它工作吗？ - Umang Gupta

你能调试一下 r = cv2.selectROI(img) 的值吗？同时附上 cv2.imshow() 的输出结果吗？ - ZdaR

该死！我希望我回答了那个问题 :P。 - Umang Gupta

Answer 1

cv2.selectROI函数返回一个矩形的(x,y,w,h)值，类似于cv2.boundingRect()函数。我猜保存的黑色矩形是由于将边界框坐标转换为int类型时出现了舍入问题。因此，直接解包(x,y,w,h)坐标，并使用Numpy切片从中提取ROI。下面是一个最小工作示例来提取和保存ROI：

输入图像->提取ROI的程序->保存的ROI

代码：

import cv2

image = cv2.imread('1.jpg')
(x,y,w,h) = cv2.selectROI(image)
ROI = image[y:y+h, x:x+w]

cv2.imshow("ROI", ROI)
cv2.imwrite("ROI.png", ROI)
cv2.waitKey()

- nathancy