18 lines
352 B
Python
18 lines
352 B
Python
|
from robot import Robot
|
||
|
|
import cv2 as cv
|
||
|
|
|
||
|
|
from cli import print_emotions
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
class OwnRobot(Robot):
|
||
|
|
count = 0
|
||
|
|
|
||
|
|
def update(self, image, emotions):
|
||
|
|
self.count += 1
|
||
|
|
if self.count > 10:
|
||
|
|
self.count = 1
|
||
|
|
print_emotions(emotions)
|
||
|
|
cv.imwrite(os.sep.join(["img", f"ownRobot{self.count}.png"]), image)
|