Нарезка изображения с помощью findContours (cv2)

Question

Я не могу опубликовать изображение из-за репутации, поэтому вот ссылка на изображение, которое я скачал:

https://www.amvplaygrounds.co.uk/pub/media/catalog/product/a/m/amv_f4-pm-017-shapes-circle-square-rectangle-star-triangle-200mm-2-sq-3d.jpg

import cv2
import numpy as np
import matplotlib.pyplot as plt 
image = cv2.imread(r'C:\Users\User\Desktop\shapes.png')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray, 200, 255, 0)
contours, hier = cv2.findContours(thresh, cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
for cnt in contours:
    x,y,w,h = cv2.boundingRect(cnt)
    shape = thresh[y:y+h,x:x+w]
    plt.imshow(shape)

Он возвращает только один контур вместо 6как вы можете видеть на картинке.

Что я делаю не так?

Answer 1

findContours() ищет белые объекты на черном фоне, поэтому вам необходимо выполнить следующие действия:

ret, thresh = cv2.threshold(gray, 225, 255, cv2.THRESH_BINARY_INV)  

, и вам нужно найтиконтуры, как это:

im2, contours, hier = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)