From aee1a867fbc23214396f545ccffb4aa023c6d6c4 Mon Sep 17 00:00:00 2001
From: LexvanGastel <l.vangastel@studen.utwente.nl>
Date: Thu, 3 Apr 2025 10:39:25 +0200
Subject: [PATCH] Vraag 26
---
Q2main.py | 75 +++++++++++++++++++++++++++++++++++++++++++++++++------
1 file changed, 68 insertions(+), 7 deletions(-)
diff --git a/Q2main.py b/Q2main.py
index e23637a..b3159a4 100644
--- a/Q2main.py
+++ b/Q2main.py
@@ -1,6 +1,8 @@
import cv2
import matplotlib.pyplot as plt
import numpy as np
+from skimage.color import rgb2gray
+from skimage.feature import blob_log
class ImageProcessor:
def __init__(self, image_path):
@@ -45,7 +47,28 @@ class ImageProcessor:
green_mask = cv2.inRange(self.image_hsv, lower_green, upper_green)
return red_mask, yellow_mask, green_mask
-
+
+def detect_blobs(mask, min_sigma=3, max_sigma=15, num_sigma = 3, threshold=0.5):
+ mask = mask.astype(np.float64) / 255.0
+ #print('mask made')
+ blobs = blob_log(mask, min_sigma=min_sigma, max_sigma=max_sigma, num_sigma=num_sigma, threshold=threshold)
+ #print('blobs detected')
+ if len(blobs) > 0:
+ blobs[:,2] = blobs[:, 2]*np.sqrt(2)
+ return blobs
+
+def plot_detected_blobs(mask, blobs, title):
+ fig, ax = plt.subplots(figsize=(6,6))
+ ax.imshow(mask, cmap='gray')
+ ax.set_title(title)
+ ax.axis("off")
+
+ for blob in blobs:
+ y, x, r = blob
+ circle =plt.Circle((x,y), r, color = 'red', linewidth = 2, fill = False)
+ ax.add_patch(circle)
+
+ plt.show()
class Plotter:
@staticmethod
@@ -121,6 +144,36 @@ class Plotter:
axes[2].axis("off")
plt.show()
+
+ @staticmethod
+ def plot_blobs(red_mask, yellow_mask, green_mask, red_blobs, yellow_blobs, green_blobs):
+ fig, axes = plt.subplots(1, 3, figsize=(18,6))
+
+ axes[0].imshow(red_mask, cmap ='gray')
+ axes[0].set_title("Red blobs")
+ axes[0].axis("off")
+ for blob in red_blobs:
+ y, x, r = blob
+ circle = plt.Circle((x,y), r, color ='red', linewidth = 2, fill = False)
+ axes[0].add_patch(circle)
+
+ axes[1].imshow(yellow_mask, cmap ='gray')
+ axes[1].set_title("Yellow blobs")
+ axes[1].axis("off")
+ for blob in yellow_blobs:
+ y, x, r = blob
+ circle = plt.Circle((x,y), r, color ='yellow', linewidth = 2, fill = False)
+ axes[1].add_patch(circle)
+
+ axes[2].imshow(green_mask, cmap ='gray')
+ axes[2].set_title("Green blobs")
+ axes[2].axis("off")
+ for blob in green_blobs:
+ y, x, r = blob
+ circle = plt.Circle((x,y), r, color ='green', linewidth = 2, fill = False)
+ axes[2].add_patch(circle)
+
+ plt.show()
image_path = "traffic_light_image_1.png"
processor = ImageProcessor(image_path)
@@ -128,19 +181,27 @@ red, green, blue = processor.get_color_channels()
red_mask, yellow_mask, green_mask = processor.get_color_masks()
-snr_red = processor.calculate_snr(red)
-snr_green = processor.calculate_snr(green)
-snr_blue = processor.calculate_snr(blue)
+red_blobs = detect_blobs(red_mask)
+yellow_blobs = detect_blobs(yellow_mask)
+green_blobs = detect_blobs(green_mask)
+
+print(f"Red Blobs: {len(red_blobs)}")
+print(f"Yellow Blobs: {len(yellow_blobs)}")
+print(f"Green Blobs: {len(green_blobs)}")
+
+# snr_red = processor.calculate_snr(red)
+# snr_green = processor.calculate_snr(green)
+# snr_blue = processor.calculate_snr(blue)
#print(f"SNR for Red Channel {snr_red:.2f}")
#print(f"SNR for Green Channel {snr_green:.2f}")
#print(f"SNR for Blue Channel {snr_blue:.2f}")
-filtered_image = processor.apply_median_filter(kernel_size = 5)
+# filtered_image = processor.apply_median_filter(kernel_size = 5)
#Plotter.plot_image_channels(processor.image_rgb,red,green,blue)
#Plotter.plot_histograms(red, green, blue)
#Plotter.plot_filtered_image(processor.image_rgb, filtered_image)
-Plotter.plot_color_masks(processor.image_rgb, red_mask, yellow_mask, green_mask)
-print(processor.image.shape)
\ No newline at end of file
+#Plotter.plot_color_masks(processor.image_rgb, red_mask, yellow_mask, green_mask)
+Plotter.plot_blobs(red_mask, yellow_mask, green_mask, red_blobs, yellow_blobs, green_blobs)
--
GitLab