gistfile1.txt

import time
from datetime import datetime

import telebot
import cv2
import numpy as np
import os

bot = telebot.TeleBot("741780913:AAHe-7xLu_z4KpvPduO8lZLc0Etz_36K_NM")

recognizer = cv2.face.LBPHFaceRecognizer_create()
recognizer.read("trainer/trainer.yml")
cascadePath = "haarcascade_frontalface_default.xml"
faceCascade = cv2.CascadeClassifier(cascadePath)

font = cv2.FONT_HERSHEY_SIMPLEX

# iniciate id counter
id = 0

# names related to ids: example ==> Marcelo: id=1,  etc
names = ["None", "Ryan", "Elon", "Ilza", "Kelvin", "W"]

# Initialize and start realtime video capture
cam = cv2.VideoCapture(0)
cam.set(3, 640)  # set video widht
cam.set(4, 480)  # set video height

# Define min window size to be recognized as a face
minW = 0.1 * cam.get(3)
minH = 0.1 * cam.get(3)

cap = cv2.VideoCapture(0)
if cap.isOpened() == False:
    print("Unable to read camera feed")

frame_width = int(cam.get(3))
frame_height = int(cam.get(4))

# Define the codec and create VideoWriter object.The output is stored in 'outpy.avi' file.
video_out = cv2.VideoWriter('outpy.avi', cv2.VideoWriter_fourcc('M','J','P','G'), 10, (frame_width, frame_height))

while True:

    # bot.polling()
    ret, img = cam.read()
    img = cv2.flip(img, -1)  # Flip vertically

    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    faces = faceCascade.detectMultiScale(
        gray, scaleFactor=1.2, minNeighbors=5, minSize=(int(minW), int(minH))
    )

    for (x, y, w, h) in faces:

        cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)

        id, confidence = recognizer.predict(gray[y : y + h, x : x + w])

        # Check if confidence is less them 100 ==> "0" is perfect match
        if confidence < 100:
            id = names[id]
            confidence = "  {0}%".format(round(100 - confidence))
        else:
            id = "unknown"
            confidence = "  {0}%".format(round(100 - confidence))
            '''
            timestamp = datetime.now()
            filename = "capture/{}.jpg".format(timestamp)
            cv2.imwrite(filename, img)
            photo = open(filename, "rb")
            bot.send_photo("793430359", photo)
            '''
            video_out.write(img)
            video = open("outpy.avi", "rb")
            bot.send_video("793430359", video)

        cv2.putText(img, str(id), (x + 5, y - 5), font, 1, (255, 255, 255), 2)
        cv2.putText(img, str(confidence), (x + 5, y + h - 5), font, 1, (255, 255, 0), 1)

    cv2.imshow("camera", img)
    time.sleep(10)

    k = cv2.waitKey(10) & 0xFF  # Press 'ESC' for exiting video
    if k == 27:
        break

# Do a bit of cleanup
print("\n [INFO] Exiting Program and cleanup stuff")
cam.release()
cv2.destroyAllWindows()