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

fixed cv2 error

Browse Source
This commit is contained in:
gocivici 2023-10-27 14:57:51 +03:00
parent 4be8c998b5
commit 4a73bb38ac
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@ -43,7 +43,7 @@ if cam.isOpened():
if TIMER>0: if TIMER>0:
draw.text((130, 300), str(TIMER), font=fontBIG,fill=(255,0,0,255)) draw.text((130, 300), str(TIMER), font=fontBIG,fill=(255,0,0,255))
else: else:
cv2.imwrite("Fear_{}.jpg".format(prev),img)
draw.text((30, 480), "ANALYZING...", font=fontSmall,fill=(255,0,0,255)) draw.text((30, 480), "ANALYZING...", font=fontSmall,fill=(255,0,0,255))
img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE) img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE)
@ -72,6 +72,7 @@ if cam.isOpened():
imgGray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) imgGray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
imgGray = cv2.equalizeHist(imgGray) imgGray = cv2.equalizeHist(imgGray)
cv2.imwrite('scared.jpg', imgGray) cv2.imwrite('scared.jpg', imgGray)
cv2.imwrite("Fear_{}.jpg".format(cur),img)
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)