added optional group photo
This commit is contained in:
gocivici
parent
30995b866b
commit
79e3b95613
140
main2.py
Normal file
140
main2.py
Normal file
@ -0,0 +1,140 @@
|
||||
import math
|
||||
import time
|
||||
import numpy as np
|
||||
import cv2 #pip install opencv-python ||| pip3 install opencv-contrib-python==4.4.0.46
|
||||
from gpiozero import Button
|
||||
from escpos.printer import Serial
|
||||
|
||||
from PIL import Image
|
||||
from PIL import ImageDraw
|
||||
from PIL import ImageFont
|
||||
|
||||
from deepface import DeepFace #pip install deepface
|
||||
button = Button(2)
|
||||
|
||||
cam = cv2.VideoCapture(0)
|
||||
cv2.namedWindow("webcam", cv2.WND_PROP_FULLSCREEN)
|
||||
cv2.setWindowProperty("webcam",cv2.WND_PROP_FULLSCREEN,cv2.WINDOW_FULLSCREEN)
|
||||
printer = Serial(devfile='/dev/serial0',baudrate=19200,bytesize=8,parity='N',stopbits=1.00,dsrdtr=True)
|
||||
# printer.set(density=10)
|
||||
cameraMode = False
|
||||
TIMER = 5
|
||||
|
||||
startScreen = cv2.imread("noFace.png")
|
||||
|
||||
if cam.isOpened():
|
||||
while True:
|
||||
ret, img = cam.read()
|
||||
# img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
|
||||
if cameraMode and ret:
|
||||
prev = time.time()
|
||||
while TIMER >= 0:
|
||||
ret, img = cam.read()
|
||||
# img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
|
||||
# cv2.putText(img, str(TIMER), (200, 250), cv2.FONT_HERSHEY_SIMPLEX, 7, (0, 255, 255), 4, cv2.LINE_AA)
|
||||
img = cv2.flip(img, 0)
|
||||
img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE)
|
||||
img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
|
||||
img = Image.fromarray(img)
|
||||
draw = ImageDraw.Draw(img)
|
||||
|
||||
fontBIG = ImageFont.truetype("HalloweenFont.ttf", 350)
|
||||
fontSmall = ImageFont.truetype("HalloweenFont.ttf", 60)
|
||||
if TIMER>0:
|
||||
draw.text((130, 150), str(TIMER), font=fontBIG,fill=(255,0,0,255))
|
||||
else:
|
||||
|
||||
draw.text((30, 480), "PRINTING...", font=fontSmall,fill=(255,0,0,255))
|
||||
img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
|
||||
img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
|
||||
|
||||
cv2.imshow('webcam',img)
|
||||
# print(str(TIMER))
|
||||
cur = time.time() #current time
|
||||
if cur-prev >= 1:
|
||||
prev = cur
|
||||
TIMER = TIMER-1
|
||||
key = cv2.waitKey(5) & 0xFF
|
||||
if ord('q') == key:
|
||||
break
|
||||
else:
|
||||
ret, img = cam.read()
|
||||
img = cv2.flip(img, 0)
|
||||
img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE)
|
||||
# img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
|
||||
try:
|
||||
predictions = DeepFace.analyze(img,actions=['emotion'])
|
||||
fearValue = predictions[0]["emotion"]["fear"]
|
||||
surpriseValue = predictions[0]["emotion"]["surprise"]
|
||||
fearPoint = max(fearValue,surpriseValue)
|
||||
print("FEAR:" + str(round(fearPoint,2)))
|
||||
print("SURPRISE:" + str(round(surpriseValue,2)))
|
||||
imgGray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
||||
imgGray = cv2.equalizeHist(imgGray)
|
||||
cv2.imwrite('scared.jpg', imgGray)
|
||||
cv2.imwrite("/home/gogo/Desktop/Photos/Fear_{}.jpg".format(cur),img)
|
||||
img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
|
||||
img = Image.fromarray(img)
|
||||
draw = ImageDraw.Draw(img)
|
||||
font_size = 75
|
||||
font = ImageFont.truetype("HalloweenFont.ttf", font_size)
|
||||
# text = "FEAR LEVEL"
|
||||
# draw.text((30, 452), str(text), font=font,fill=(255,0,0,255))
|
||||
img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
|
||||
|
||||
# cv2.rectangle(img,(30,550),(30+math.floor(int(fearPoint)*420/100),600),(255,255,255), -1)
|
||||
# cv2.rectangle(img,(30,550),(450,600),(0,0,255), 8)
|
||||
img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
|
||||
|
||||
cv2.imshow('webcam',img)
|
||||
cv2.waitKey(2000)
|
||||
if fearPoint>0:
|
||||
#rotoImg = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
|
||||
# cv2.imshow('webcam',img)
|
||||
basewidth = 384
|
||||
imgCrop = Image.open('scared.jpg')
|
||||
wpercent = (basewidth/float(imgCrop.size[0]))
|
||||
hsize = int((float(imgCrop.size[1])*float(wpercent)))
|
||||
imgCrop = imgCrop.resize((basewidth,hsize), Image.Resampling.LANCZOS)
|
||||
imgCrop = imgCrop.save("cropScared.jpg")
|
||||
# cv2.waitKey(2000)
|
||||
printer.set(align='center',font='a',width=2,height=2)
|
||||
|
||||
printer.image("cropScared.jpg",high_density_vertical=True,high_density_horizontal=False,impl="bitImageRaster")
|
||||
printer.text("HALLOWEEN 2023\n")
|
||||
# printer.text("Surprise Level: \n" + str(round(surpriseValue,2))+"/100\n")
|
||||
# printer.text("Overall: \n" + str(round(fearPoint,2))+"/100\n")
|
||||
# if fearPoint>80:
|
||||
# printer.text("(Scream Queen)\n")
|
||||
# if fearPoint<5:
|
||||
# printer.text("(The Killer)\n")
|
||||
printer.text("\n\n\n\n")
|
||||
#printer.set(align='center',font='b',width=1,height=1)
|
||||
#printer.text("Spooky Night 2023")
|
||||
#printer.text("2023\n")
|
||||
# cv2.waitKey(5000)
|
||||
except:
|
||||
cameraMode = False
|
||||
TIMER = 5
|
||||
|
||||
#print(30+math.floor(int(fearPoint)*580/100))
|
||||
# ft.putText(img=img,text='TEST',org=(15, 70),fontHeight=60,color=(255, 255, 255),thickness=-1,line_type=cv2.LINE_AA,bottomLeftOrigin=True)
|
||||
#cv2.rectangle(img,(30,400),(610,450),(255,255,255), 5)
|
||||
#cv2.rectangle(img,(30,400),(30+math.floor(int(fearPoint)*580/100),450),(255,255,255), -1)
|
||||
|
||||
# cv2.waitKey(5000)
|
||||
|
||||
|
||||
|
||||
cameraMode = False
|
||||
TIMER = 5
|
||||
else:
|
||||
cv2.imshow('webcam',startScreen)
|
||||
|
||||
key = cv2.waitKey(5) & 0xFF
|
||||
if ord('t') == key or button.is_pressed:
|
||||
cameraMode = True
|
||||
if ord('q') == key:
|
||||
break
|
||||
|
||||
cam.release()
|
||||
Loading…
Reference in New Issue
Block a user