我想裁剪到图像中的对象,以便只保留彩色对象。我怎样才能在 python 中以最有效的方式做到这一点?
基本上,图像具有黑色(0,0,0(背景,但对象的颜色不同。我想裁剪到对象以删除无用的背景。
我知道 cv2 具有 resize(( 函数,但他们无法检测到它是否是背景。我也可以循环整个图像以找到位置,但这太慢了。
最后,
我找到了一个 API 来完成这项工作。
使用cv2.findContours(( 从遮罩图像(具有相应颜色的对象(中获取对象的位置,并直接使用 numpy 进行剪切。
def cut_object(rgb_image,mask_image,object_color):
"""This function is used to cut a specific object from the pair RGB/mask image."""
rgb_image=cv2.imread(rgb_image)
mask_image=cv2.imread(mask_image)
mask_image=cv2.cvtColor(mask_image,cv2.COLOR_BGR2RGB)
# Create mask image with the only object
object_mask_binary=cv2.inRange(mask_image,object_color,object_color)
object_mask=cv2.bitwise_and(mask_image,mask_image,mask=object_mask_binary)
# Detect the position of the object
object_contour=cv2.cvtColor(object_mask,cv2.COLOR_BGR2GRAY)
object_position,c=cv2.findContours(object_contour,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
object_position=np.array(object_position).squeeze()
hmin,hmax=object_position[:,:1].min(),object_position[:,:1].max()
wmin,wmax=object_position[:,1:2].min(),object_position[:,1:2].max()
# Cut the object from the RGB image
crop_rgb=rgb_image[wmin:wmax,hmin:hmax]
return crop_rgb