img_segmentation.py

import cv2
import numpy as np
import matplotlib.pyplot as plt

# Read image
img = cv2.imread("photo.jpg")
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

# -------------------------
# RGB Segmentation
# Segment RED color (example)
# -------------------------
lower_rgb = np.array([150, 0, 0])
upper_rgb = np.array([255, 80, 80])
mask_rgb = cv2.inRange(img_rgb, lower_rgb, upper_rgb)
seg_rgb = cv2.bitwise_and(img_rgb, img_rgb, mask=mask_rgb)

# -------------------------
# HSV Segmentation
# Segment RED color in HSV
# -------------------------
img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

lower_hsv = np.array([0, 120, 70])
upper_hsv = np.array([10, 255, 255])

mask_hsv = cv2.inRange(img_hsv, lower_hsv, upper_hsv)
seg_hsv = cv2.bitwise_and(img, img, mask=mask_hsv)

# Display Results
plt.figure(figsize=(12, 8))

plt.subplot(2, 2, 1)
plt.title("Original")
plt.imshow(img_rgb)
plt.axis('off')

plt.subplot(2, 2, 2)
plt.title("RGB Segmentation")
plt.imshow(seg_rgb)
plt.axis('off')

plt.subplot(2, 2, 3)
plt.title("HSV Segmentation")
plt.imshow(cv2.cvtColor(seg_hsv, cv2.COLOR_BGR2RGB))
plt.axis('off')

plt.tight_layout()
plt.show()