Я думаю, что одной из проблем может быть построение круга (координаты могут быть неправильными).
Кроме того, как упоминалось в @Nicos, есть много настроек, связанных с традиционной обработкой изображений, чтобы заставить конкретные случаи работать (в то время как более поздние подходы к машинному обучению - это то, что модели не перетренируются), моя попытка с лыжным магом отображается ниже. Диапазон радиуса, количество кругов, изображение для обнаружения краев - все это нужно настроить ... учитывая потенциальные различия между изображениями и внутри них. В этом изображении, по крайней мере для меня, есть 3 круга с переменным градиентом, из изображения canny
обнаружения краев, вы можете видеть, что мы получаем более 3 кругов, кроме того, "освещенность", кажется, меняется на в разных местах (из-за того, что это sem-изображение)?!
import matplotlib.pyplot as plt
import numpy as np
import imageio
from skimage import data, color
from skimage.transform import hough_circle, hough_circle_peaks
from skimage.feature import canny
from skimage.draw import circle_perimeter
from skimage.util import img_as_ubyte
!wget https://i.stack.imgur.com/2tsWw.jpg
# rgb to gray https://stackoverflow.com/a/51571053/868736
im = imageio.imread('2tsWw.jpg')
gray = lambda rgb : np.dot(rgb[... , :3] , [0.299 , 0.587, 0.114])
gray = gray(im)
image = np.array(gray[60:220,210:450])
plt.imshow(image,cmap='gray')
edges = canny(image, sigma=3,)
plt.imshow(edges,cmap='gray')
overlayimage = np.copy(image)
# https://scikit-image.org/docs/dev/auto_examples/edges/plot_circular_elliptical_hough_transform.html
hough_radii = np.arange(30, 60, 2)
hough_res = hough_circle(edges, hough_radii)
# Select the most prominent X circles
x=1
accums, cx, cy, radii = hough_circle_peaks(hough_res, hough_radii,
total_num_peaks=x)
# Draw them
fig, ax = plt.subplots(ncols=1, nrows=1, figsize=(10, 4))
#image = color.gray2rgb(image)
for center_y, center_x, radius in zip(cy, cx, radii):
circy, circx = circle_perimeter(center_y, center_x, radius)
overlayimage[circy, circx] = 255
print(radii)
ax.imshow(overlayimage,cmap='gray')
plt.show()