20 lines
595 B
Python
20 lines
595 B
Python
import cv2 as cv
|
|
import numpy as np
|
|
|
|
|
|
def mean_shift_segmentation(image, spatial_radius, color_radius):
|
|
image_yuv = cv.cvtColor(image, cv.COLOR_BGR2YUV)
|
|
segmented_image = cv.pyrMeanShiftFiltering(image_yuv, spatial_radius, color_radius)
|
|
return cv.cvtColor(segmented_image, cv.COLOR_YUV2BGR)
|
|
|
|
|
|
image = cv.imread("images/lena-color.png")
|
|
spatial_radius = 10
|
|
color_radius = 10
|
|
segmented_image = mean_shift_segmentation(image, spatial_radius, color_radius)
|
|
|
|
cv.imshow("Original Image", image)
|
|
cv.imshow("Mean Shift Segmented Image", segmented_image)
|
|
cv.waitKey(0)
|
|
cv.destroyAllWindows()
|