xxxxxxxxxx
147
let mobilenet;
let video;
let detections = [];
let eyes = []; // Array zum Speichern der Augenpositionen und -größen
let lastPersonCount = 0;
function modelReady() {
console.log('Model is ready!!!');
mobilenet.detect(gotResults);
}
function gotResults(error, results) {
if (error) {
console.error(error);
} else {
detections = results;
mobilenet.detect(gotResults);
}
}
function setup() {
createCanvas(800, 600);
video = createCapture(VIDEO);
video.size(800, 600);
video.hide();
mobilenet = ml5.objectDetector('cocossd', video, modelReady);
}
function draw() {
background(220);
// Zeichne das Video und erkenne Personen
image(video, 0, 0);
let currentPersonCount = 0;
let newEyes = [];
if (detections) {
for (let i = 0; i < detections.length; i++) {
let object = detections[i];
if (object.label === 'person') {
let x = object.x + object.width / 2; // Mitte des Rechtecks
let y = object.y + object.height / 2; // Mitte des Rechtecks
let diameter = min(object.width, object.height) / 2; // Durchmesser basierend auf der kleineren Seite des Rechtecks
newEyes.push({ x: x, y: y, diameter: diameter });
noFill();
stroke(255, 0, 0);
strokeWeight(2);
rect(object.x, object.y, object.width, object.height);
currentPersonCount++;
}
}
fill(255);
textSize(20);
text('Anzahl der Personen: ' + currentPersonCount, 10, 30);
} else {
fill(255);
textSize(32);
textAlign(CENTER, CENTER);
text('Lade...', width / 2, height / 2);
}
// Wenn sich die Anzahl der erkannten Personen ändert, aktualisiere die Augenpositionen
if (currentPersonCount !== lastPersonCount) {
eyes = placeEyes(currentPersonCount, newEyes);
lastPersonCount = currentPersonCount;
}
// Zeichne die Augen
drawEyes(newEyes);
}
function placeEyes(personCount, newEyes) {
let placedEyes = [];
for (let i = 0; i < personCount; i++) {
let eye = newEyes[i];
// Versuche eine Position zu finden, bei der sich die schwarzen Kreise nicht überlappen
let fixedX, fixedY;
let tries = 0;
let maxTries = 100;
do {
fixedX = random(eye.diameter / 2, width - eye.diameter / 2);
fixedY = random(eye.diameter / 2, height - eye.diameter / 2);
let overlapping = false;
for (let j = 0; j < placedEyes.length; j++) {
let otherEye = placedEyes[j];
let distance = dist(fixedX, fixedY, otherEye.x, otherEye.y);
if (distance < (eye.diameter + otherEye.diameter) / 2) {
overlapping = true;
break;
}
}
if (!overlapping) {
placedEyes.push({ x: fixedX, y: fixedY, diameter: eye.diameter });
break;
}
tries++;
} while (tries < maxTries);
}
return placedEyes;
}
function drawEyes(newEyes) {
for (let i = 0; i < eyes.length; i++) {
let eye = eyes[i];
let person = newEyes[i];
// Zeichne den äußeren schwarzen Kreis
fill(0);
noStroke();
ellipse(eye.x, eye.y, eye.diameter, eye.diameter);
// Berechne die Position des inneren weißen Kreises basierend auf der Position der Person
let innerDiameter = eye.diameter / 2;
let maxOffset = (eye.diameter - innerDiameter) / 2;
// Berechne den Abstand zwischen dem Mittelpunkt des schwarzen Kreises und der Position der Person
let distFromCenter = dist(person.x, person.y, eye.x, eye.y);
// Setze den weißen Kreis innerhalb des schwarzen Kreises basierend auf der Position der Person
let angle = atan2(person.y - eye.y, person.x - eye.x);
let whiteX = eye.x + cos(angle) * maxOffset;
let whiteY = eye.y + sin(angle) * maxOffset;
fill(255);
ellipse(whiteX, whiteY, innerDiameter, innerDiameter);
// Berechne die Position des inneren blauen Kreises basierend auf der Position des weißen Kreises
let blueDiameter = innerDiameter * 0.7; // Der Durchmesser des blauen Kreises ist 70% des Durchmessers des weißen Kreises
let blueMaxOffset = (innerDiameter - blueDiameter) / 2;
let blueOffsetX = constrain(whiteX - eye.x, -blueMaxOffset, blueMaxOffset);
let blueOffsetY = constrain(whiteY - eye.y, -blueMaxOffset, blueMaxOffset);
fill(0, 0, 255);
ellipse(whiteX + blueOffsetX, whiteY + blueOffsetY, blueDiameter, blueDiameter);
}
}