From 0444ca933fbb08ebfb892b313972b42064193519 Mon Sep 17 00:00:00 2001
From: Peter-Simon Dieterich <dieterich.peter@gmail.com>
Date: Fri, 1 Nov 2019 11:49:19 +0100
Subject: [PATCH] Fix missing colors in overlay

---
 src/cli/test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/cli/test.py b/src/cli/test.py
index 16fadb1..02a557c 100644
--- a/src/cli/test.py
+++ b/src/cli/test.py
@@ -124,6 +124,7 @@ try:
 		frame = clahe.apply(frame)
 		# Make a frame to put overlays in
 		overlay = frame.copy()
+		overlay = cv2.cvtColor(overlay, cv2.COLOR_GRAY2BGR)
 
 		# Fetch the frame height and width
 		height, width = frame.shape[:2]
@@ -190,6 +191,7 @@ try:
 
 		# Add the overlay to the frame with some transparency
 		alpha = 0.65
+		frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2BGR)
 		cv2.addWeighted(overlay, alpha, frame, 1 - alpha, 0, frame)
 
 		# Show the image in a window