Как считать бриллианты на ювелирных изделиях лучше

Question

У меня есть изображение ниже на драгоценном камне с бриллиантами на нем, и я пытаюсь обработать изображение на нем:

Можно ли обнаружить квадрат? объекты и показать их с контурной квадратной формой?

Мой пример кода:

 shifted = cv2.pyrMeanShiftFiltering(image, 21, 51)
self.gray = cv2.cvtColor(shifted, cv2.COLOR_BGR2GRAY)
self.gray_image = copy.copy(self.gray)

thresh = threshold_yen(self.gray)
if not blackCheck:
    thresh = self.gray > thresh
else:
    thresh = self.gray < thresh
D = ndimage.distance_transform_edt(thresh)
localMax = peak_local_max(D, indices=False, min_distance=11, labels=thresh)
markers = ndimage.label(localMax, structure=np.ones((3, 3)))[0]
self.labels = watershed(-D, markers, mask=thresh)