How can I use Python for image segmentation?

How can I use Python for image segmentation?

You can use libraries like OpenCV, scikit-image, or TensorFlow for image segmentation in Python. Would you like to know about a specific library?

Tell me more about using OpenCV for segmentation.

OpenCV provides various methods like thresholding, contour detection, and watershed segmentation. Do you want a code example?

Yes, please provide a code example for watershed segmentation.

Here's a simple example:

import cv2
import numpy as np

# Load image
image = cv2.imread('image.jpg')

# Convert to grayscale
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# Apply thresholding
_, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)

# Find sure background area
kernel = np.ones((3, 3), np.uint8)

# Dilate the thresholded image
sure_bg = cv2.dilate(thresh, kernel, iterations=3)

# Find sure foreground area
contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# Create markers
markers = np.zeros_like(gray)
for i in range(len(contours)):
    cv2.drawContours(markers, contours, i, i + 1, -1)

# Apply watershed algorithm
markers = cv2.watershed(image, markers)
image[markers == -1] = [255, 0, 0]  # Mark boundaries in red

# Show result
cv2.imshow('Segmented Image', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

This code segments an image using the watershed algorithm. Would you like to explore more techniques?