Gorkem/FearBooth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

edit interface

Browse Source
This commit is contained in:
gocivici 2023-10-16 16:16:38 +03:00
parent 30b5991690
commit 088f772b5f
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
main.py
View File

@ -13,8 +13,8 @@ max_samples = 10
# img = cv2.imread("test.jpg") # img = cv2.imread("test.jpg")
cam = cv2.VideoCapture(1) cam = cv2.VideoCapture(1)
# cam.set(cv2.cv.CV_CAP_PROP_FPS, 10) # cam.set(cv2.cv.CV_CAP_PROP_FPS, 10)
cam.set(cv2.CAP_PROP_FRAME_HEIGHT, 240) # cam.set(cv2.CAP_PROP_FRAME_HEIGHT, 240)
cam.set(cv2.CAP_PROP_FRAME_WIDTH, 320) # cam.set(cv2.CAP_PROP_FRAME_WIDTH, 320)
noFace = cv2.imread("noFace.png") noFace = cv2.imread("noFace.png")
# ft = cv2.freetype.createFreeType2() # ft = cv2.freetype.createFreeType2()
@ -30,10 +30,10 @@ if cam.isOpened():
while True: while True:
ret, img = cam.read() ret, img = cam.read()
if ret: if ret:
img = cv2.resize(img, (80,60), interpolation = cv2.INTER_AREA) face = cv2.resize(img, (120,160), interpolation = cv2.INTER_AREA)
print(img.shape) #print(img.shape)
try: try:
predictions = DeepFace.analyze(img,actions=['emotion']) predictions = DeepFace.analyze(face,actions=['emotion'])
fearPoint = predictions[0]["emotion"]["fear"] fearPoint = predictions[0]["emotion"]["fear"]
readings = np.append(readings, fearPoint) readings = np.append(readings, fearPoint)
@ -47,18 +47,18 @@ if cam.isOpened():
# avgNumber = avgNumber + fearPoint # avgNumber = avgNumber + fearPoint
# print(avgNumber) # print(avgNumber)
# img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB) img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
# img = Image.fromarray(img) img = Image.fromarray(img)
# draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
# font_size = 65 font_size = 65
# font = ImageFont.truetype("HalloweenFont.ttf", font_size) font = ImageFont.truetype("HalloweenFont.ttf", font_size)
# text = "FEAR LEVEL" text = "FEAR LEVEL"
# draw.text((144, 308), str(text), font=font,fill=(255,0,0,255)) draw.text((144, 308), str(text), font=font,fill=(255,0,0,255))
# img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
#print(30+math.floor(int(fearPoint)*580/100)) #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) # 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),(610,450),(255,255,255), 5)
# cv2.rectangle(img,(30,400),(30+math.floor(int(avg)*580/100),450),(255,255,255), -1) cv2.rectangle(img,(30,400),(30+math.floor(int(avg)*580/100),450),(255,255,255), -1)
cv2.imshow('webcam',img) cv2.imshow('webcam',img)
except Exception as e: except Exception as e:
print(e) print(e)