Compare commits
4
Commits
27039171cf
..
master
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
f554f503c9 | ||
|
|
ad2eea06e5 | ||
|
|
ddea6ea320 | ||
|
|
4524a0a32b |
No files matched your search
+1
-1
@@ -1,3 +1,3 @@
|
||||
Bilal IP : 192.168.50.130
|
||||
Bilal: 192.168.50.130
|
||||
Youssef: 192.168.50.226
|
||||
David: 192.168.50.79
|
||||
@@ -1,153 +0,0 @@
|
||||
import cv2
|
||||
import numpy as np
|
||||
from game import Game
|
||||
# from Track import nothing
|
||||
# Farbwerte für die Erkennung (Beispiel: Rot, Grün, Blau)
|
||||
colors = [(0, 0, 255), (0, 255, 0), (255, 0, 0)]
|
||||
color_names = ["Rot", "Gruen", "Blau"]
|
||||
|
||||
# Farbgrenzen für die Erkennung
|
||||
lower_red = np.array([0, 100, 100])
|
||||
upper_red = np.array([10, 255, 255])
|
||||
|
||||
lower_green = np.array([40, 100, 100])
|
||||
upper_green = np.array([70, 255, 255])
|
||||
|
||||
lower_blue = np.array([90, 100, 100])
|
||||
upper_blue = np.array([130, 255, 255])
|
||||
|
||||
|
||||
CURRENT_SCORES= {'score_red': 0,
|
||||
'score_green': 0,
|
||||
'score_blue': 0
|
||||
}
|
||||
|
||||
# Funktion zur Farberkennung
|
||||
def erkennung_farben(img):
|
||||
hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
|
||||
results = []
|
||||
|
||||
count_red = 0
|
||||
count_green = 0
|
||||
count_blue = 0
|
||||
|
||||
for i, color in enumerate(colors):
|
||||
if i == 0:
|
||||
lower = lower_red
|
||||
upper = upper_red
|
||||
elif i == 1:
|
||||
lower = lower_green
|
||||
upper = upper_green
|
||||
elif i == 2:
|
||||
lower = lower_blue
|
||||
upper = upper_blue
|
||||
|
||||
mask = cv2.inRange(hsv_img, lower, upper)
|
||||
|
||||
# Farbfläche finden
|
||||
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
|
||||
center = None
|
||||
count = 0
|
||||
|
||||
for contour in contours:
|
||||
if count < 3:
|
||||
if cv2.contourArea(contour) > 100:
|
||||
# Schwerpunkt der Kontur berechnen
|
||||
M = cv2.moments(contour)
|
||||
if M["m00"] > 0:
|
||||
cX = int(M["m10"] / M["m00"])
|
||||
cY = int(M["m01"] / M["m00"])
|
||||
center = (cX, cY)
|
||||
count += 1
|
||||
# Rechteck zeichnen
|
||||
x, y, w, h = cv2.boundingRect(contour)
|
||||
cv2.rectangle(img, (x, y), (x + w, y + h), color, 2)
|
||||
cv2.putText(img, color_names[i], (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, color, 2)
|
||||
|
||||
# Farbanzahl erhöhen
|
||||
if i == 0:
|
||||
count_red += 1
|
||||
elif i == 1:
|
||||
count_green += 1
|
||||
elif i == 2:
|
||||
count_blue += 1
|
||||
|
||||
|
||||
|
||||
results.append(center)
|
||||
|
||||
return img, results, count_red, count_green, count_blue
|
||||
|
||||
# Funktion zur Positionsermittlung
|
||||
def ermittle_position(results, img_width):
|
||||
positions = []
|
||||
|
||||
for result in results:
|
||||
if result is None:
|
||||
position = "Nicht gefunden"
|
||||
else:
|
||||
x = result[0]
|
||||
if x < img_width / 3:
|
||||
position = "Rechts"
|
||||
elif x < 2 * img_width / 3:
|
||||
position = "Mitte"
|
||||
else:
|
||||
position = "Links"
|
||||
|
||||
positions.append(position)
|
||||
|
||||
return positions
|
||||
|
||||
def count_colors(red: int, green: int, blue: int) -> None:
|
||||
currrent_scores = {'score_red': red,
|
||||
'score_green': green,
|
||||
'score_blue': blue
|
||||
}
|
||||
Game.set_scoreboard(currrent_scores)
|
||||
|
||||
# Hauptprogramm
|
||||
if __name__ == "__main__":
|
||||
# Videoquelle öffnen (kann auch eine Bilddatei sein)
|
||||
video = cv2.VideoCapture(1) # Hier "0" für die Kamera verwenden
|
||||
|
||||
while True:
|
||||
# Einzelbild von der Videoquelle lesen
|
||||
ret, frame = video.read()
|
||||
|
||||
# Fehlerbehandlung, wenn kein Bild gelesen werden kann
|
||||
if not ret:
|
||||
break
|
||||
# Farben erkennen
|
||||
farben_img, ergebnisse, count_red, count_green, count_blue = erkennung_farben(frame)
|
||||
|
||||
# Add counted colours to the scoreboard in the correct field
|
||||
count_colors(count_red, count_green, count_blue)
|
||||
|
||||
# Anzahl der Farben anzeigen
|
||||
cv2.putText(farben_img, f"Rot: {count_red}", (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, colors[0], 2)
|
||||
cv2.putText(farben_img, f"Gruen: {count_green}", (10, 60), cv2.FONT_HERSHEY_SIMPLEX, 1, colors[1], 2)
|
||||
cv2.putText(farben_img, f"Blau: {count_blue}", (10, 90), cv2.FONT_HERSHEY_SIMPLEX, 1, colors[2], 2)
|
||||
|
||||
# Positionen ermitteln
|
||||
img_width = frame.shape[1]
|
||||
positionen = ermittle_position(ergebnisse, img_width)
|
||||
|
||||
# Positionen anzeigen
|
||||
for i, position in enumerate(positionen):
|
||||
cv2.putText(farben_img, f"{color_names[i]}: {position}", (10, 150 + 30 * i),
|
||||
cv2.FONT_HERSHEY_SIMPLEX, 1, colors[i], 2)
|
||||
|
||||
# Linien zeichnen
|
||||
cv2.line(farben_img, (img_width // 3, 0), (img_width // 3, frame.shape[0]), (0, 0, 0), 2)
|
||||
cv2.line(farben_img, (2 * img_width // 3, 0), (2 * img_width // 3, frame.shape[0]), (0, 0, 0), 2)
|
||||
|
||||
# Bild anzeigen
|
||||
cv2.imshow("Farberkennung", farben_img)
|
||||
|
||||
# Auf "q" drücken, um die Schleife zu beenden
|
||||
if cv2.waitKey(1) & 0xFF == ord('q'):
|
||||
break
|
||||
|
||||
# Videoquelle und Fenster schließen
|
||||
video.release()
|
||||
cv2.destroyAllWindows()
|
||||
@@ -1,7 +1,7 @@
|
||||
import time
|
||||
import numpy as np
|
||||
import cv2
|
||||
|
||||
|
||||
cap = cv2.VideoCapture(0)
|
||||
|
||||
def nothing(x):
|
||||
|
||||
@@ -16,7 +16,7 @@ questions = [('Wie heißt der längste Fluss auf der Welt?', 'Amazonas', 'Nil',
|
||||
('Wie heißt der höchste Berg Europas?', 'Zugspitze', 'Matterhorn', 'Montblanc', 'Montblanc'),
|
||||
('Wie lange braucht das Licht in etwa von der Sonne zur Erde?', '80 Sekunden', '8 Minuten', '7 Tage', '8 Minuten'),
|
||||
('Was hat James Watt erfunden?', 'Glühbirne', 'Elektrischen Widerstand', 'Dampfmaschine', 'Dampfmaschine'),
|
||||
('Wie hieß der erste Deutsch Bundeskanzler?', 'Helmut Schmidt', 'Konrad Adenauer', 'Willy Brandt', 'Konrad Adenauer'),
|
||||
('Wie hieß der erste Deutsche Bundeskanzler?', 'Helmut Schmidt', 'Konrad Adenauer', 'Willy Brandt', 'Konrad Adenauer'),
|
||||
('Von wem stammt das Gemälde der Mona Lisa?', 'Picasso', 'Vincent van Gogh', 'Leonardo da Vinci', 'Leonardo da Vinci'),
|
||||
('Wie heißt der Erfinder der Jeanshose?', 'Tom Wrangler', 'Levi Strauss', 'Peter Diesel', 'Levi Strauss'),
|
||||
('Welches Land gewann 2014 die Fußball Weltmeisterschaft?', 'Brasilien', 'Argentien', 'Deutschland', 'Deutschland'),
|
||||
|
||||
@@ -0,0 +1,129 @@
|
||||
import cv2
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
|
||||
|
||||
|
||||
class Camera():
|
||||
def __init__(self) -> None:
|
||||
self.colors = [(0, 0, 255), (0, 255, 0), (255, 0, 0)]
|
||||
self.color_names = ["Rot", "Gruen", "Blau"]
|
||||
|
||||
self.lower_red = np.array([0, 185, 15])
|
||||
self.upper_red = np.array([50, 255, 255])
|
||||
|
||||
self.lower_green = np.array([20, 70, 25])
|
||||
self.upper_green = np.array([65, 200, 255])
|
||||
|
||||
self.lower_blue = np.array([100, 220, 5])
|
||||
self.upper_blue = np.array([255, 255, 255])
|
||||
|
||||
self.video = cv2.VideoCapture(0)
|
||||
self.start_process = True
|
||||
self.correct_field_frame = 0
|
||||
self.mean_error = []
|
||||
self.scores = {'score_red': 0,
|
||||
'score_green': 0,
|
||||
'score_blue': 0
|
||||
}
|
||||
|
||||
def get_frame(self):
|
||||
try:
|
||||
_, self.image = self.video.read(0)
|
||||
except Exception as err:
|
||||
print("Can not capture the video..\n")
|
||||
print(err)
|
||||
return self.image
|
||||
|
||||
def take_picture(self):
|
||||
path = 'src_folder/Game_Images/'
|
||||
file_name = 'current_score.png'
|
||||
image_path = path+file_name
|
||||
cv2.imwrite(image_path, self.image)
|
||||
|
||||
def detect_color(self, image):
|
||||
hsv_img = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
|
||||
count_red, count_green, count_blue= 0,0,0
|
||||
|
||||
for i, color in enumerate(self.colors):
|
||||
if i == 0:
|
||||
lower = self.lower_red
|
||||
upper = self.upper_red
|
||||
elif i == 1:
|
||||
lower = self.lower_green
|
||||
upper = self.upper_green
|
||||
elif i == 2:
|
||||
lower = self.lower_blue
|
||||
upper = self.upper_blue
|
||||
|
||||
mask = cv2.inRange(hsv_img, lower, upper)
|
||||
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
|
||||
|
||||
for contour in contours:
|
||||
if 100 < cv2.contourArea(contour):
|
||||
x, y, w, h = cv2.boundingRect(contour)
|
||||
cv2.rectangle(image, (x, y), (x + w, y + h), color, 2)
|
||||
cv2.putText(image, self.color_names[i], (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, color, 2)
|
||||
|
||||
if i == 0:
|
||||
count_red += 1
|
||||
elif i == 1:
|
||||
count_green += 1
|
||||
elif i == 2:
|
||||
count_blue += 1
|
||||
|
||||
self.scores['score_red'] = count_red
|
||||
self.scores['score_green'] = count_green
|
||||
self.scores['score_blue'] = count_blue
|
||||
|
||||
def current_score(self, scores: dict):
|
||||
return scores
|
||||
|
||||
def range_of_interest(self, frame, correct_field: int):
|
||||
num_windows_in_y = 1
|
||||
num_windows_in_x = 3
|
||||
|
||||
height, width, _ = frame.shape
|
||||
roi_height = height/num_windows_in_y
|
||||
roi_width = width/num_windows_in_x
|
||||
images = []
|
||||
|
||||
for x in range(0,num_windows_in_y):
|
||||
for y in range(0,num_windows_in_x):
|
||||
tmp_image=frame[int(x*roi_height):int((x+1)*roi_height), int(y*roi_width):int((y+1)*roi_width)]
|
||||
images.append(tmp_image)
|
||||
|
||||
correct_field_frame = images[correct_field]
|
||||
return correct_field_frame
|
||||
|
||||
def reduce_errors(self, mean_error_list: list):
|
||||
for team_color in self.current_score.keys():
|
||||
df = pd.DataFrame(mean_error_list)
|
||||
total = df[team_color].sum()
|
||||
number_of_rows = len(df.index)
|
||||
error = total/number_of_rows
|
||||
self.current_score[team_color] = round(error)
|
||||
|
||||
def process(self):
|
||||
square_error = []
|
||||
|
||||
for i in range(0,100):
|
||||
frame = self.get_frame()
|
||||
interested_area = self.range_of_interest(frame, self.correct_field_frame)
|
||||
self.detect_color(interested_area)
|
||||
|
||||
cv2.line(img=frame, pt1=(frame.shape[1]//3, 0), pt2=(frame.shape[1]//3, frame.shape[0]), color=(0, 0, 0), thickness=2)
|
||||
cv2.line(img=frame, pt1=(2 * frame.shape[1]//3, 0), pt2=(2 * frame.shape[1]//3, frame.shape[0]), color=(0, 0, 0), thickness=2)
|
||||
cv2.imshow("Kamera 1,2 oder 3", frame)
|
||||
|
||||
square_error.append(self.scores)
|
||||
self.take_picture()
|
||||
print(f'{i}. Iteration scores: {self.scores}')
|
||||
self.start_process = False
|
||||
|
||||
|
||||
# nur zum testen: my_camera.start_process auf True setzen und correct_field_frame zwischen 1 und 3 wählen
|
||||
# if __name__ == "__main__":
|
||||
# my_camera = Camera()
|
||||
# my_camera.process()
|
||||
|
||||
@@ -2,6 +2,7 @@ from Database.database import QuestionDataBase
|
||||
import random
|
||||
|
||||
|
||||
|
||||
class Game:
|
||||
def __init__(self, teamsize=0, correct_field=0):
|
||||
self.teamsize = teamsize
|
||||
@@ -9,7 +10,6 @@ class Game:
|
||||
'score_green': 0,
|
||||
'score_blue': 0
|
||||
}
|
||||
|
||||
self.questions = QuestionDataBase('src_folder/BackEnd/Database/EinsZweiOderDrei.db')
|
||||
self.available_questions = list(range(1, self.questions.num_rows()))
|
||||
self.field = correct_field
|
||||
@@ -19,13 +19,13 @@ class Game:
|
||||
def set_teamsize(self, teamsize: int):
|
||||
self.teamsize = teamsize
|
||||
|
||||
def add_score(self, current_score: dict):
|
||||
def add_score(self, current_score: dict) -> dict:
|
||||
for key in self.scoreboard.keys():
|
||||
if key in current_score.keys():
|
||||
self.scoreboard[key] = self.scoreboard[key] + current_score[key]
|
||||
return self.scoreboard
|
||||
|
||||
def set_scoreboard(self, current_scores: dict):
|
||||
def set_scoreboard(self, current_scores: dict) -> dict:
|
||||
self.scoreboard = self.add_score(current_scores)
|
||||
return self.scoreboard
|
||||
|
||||
@@ -50,7 +50,6 @@ class Game:
|
||||
def shuffle_answeroptions(self):
|
||||
answeroptions = ['Answeroption_1', 'Answeroption_2', 'Answeroption_3']
|
||||
keys = []
|
||||
print(self.question)
|
||||
|
||||
for answer in answeroptions:
|
||||
keys.append(self.question[answer])
|
||||
@@ -73,6 +72,3 @@ class Game:
|
||||
|
||||
def final_result(self):
|
||||
self.final_score = dict(sorted(self.scoreboard.items(), key=lambda x: x[1], reverse=True))
|
||||
|
||||
|
||||
|
||||
@@ -1,9 +1,11 @@
|
||||
from flask import Flask, jsonify, Response, request
|
||||
from camera import Camera
|
||||
from game import Game
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
my_game = Game()
|
||||
my_camera = Camera()
|
||||
|
||||
@app.route('/question', methods=['GET'])
|
||||
def get_question():
|
||||
@@ -20,13 +22,17 @@ def teamsize():
|
||||
|
||||
@app.route('/scoreboard', methods=['GET'])
|
||||
def scoreboard():
|
||||
my_game.set_scoreboard()
|
||||
print(my_game.scoreboard)
|
||||
return jsonify(my_game.scoreboard)
|
||||
|
||||
@app.route('/check', methods=['GET'])
|
||||
def check():
|
||||
return my_game.scoreboard
|
||||
my_camera.start_process = True
|
||||
my_camera.correct_field_frame = my_game.field
|
||||
my_camera.process()
|
||||
my_game.set_scoreboard(my_camera.scores)
|
||||
my_camera.start_process = False
|
||||
return jsonify(my_game.scoreboard)
|
||||
|
||||
@app.route('/reset', methods=['GET'])
|
||||
def reset():
|
||||
@@ -46,14 +52,11 @@ def set_winner():
|
||||
|
||||
def main():
|
||||
## Router
|
||||
# app.run(host='192.168.50.79', port=5555, debug=True)
|
||||
app.run(host='192.168.50.79', port=5555, debug=True)
|
||||
|
||||
## Postman
|
||||
app.run(host='127.0.0.1', port=5555, debug=True)
|
||||
# app.run(host='127.0.0.1', port=5555, debug=True)
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 338 KiB |
Binary file not shown.
Reference in new issue
Block a user