def _train():
"""Train a new item."""
import shutil
from fridgeai import camera, training
frames = camera.get_frames(shape=(32, 32), count=100, interval=2)
label = input("Enter item name: ")
os.mkdir(label)
for i, frame in enumerate(frames):
cv2.imwrite(os.path.join(label, 'frame{}.jpg'.format(i)), frame)
training.send_training_data(label)
shutil.rmtree(label)
def rescanItem(self):
while True:
print("Scanning...")
item = ai.predict(camera.get_frames((32, 32), count=5, interval=5))
print("Scan result: {}".format(item))
break
self.label_3.setText(item)
self.label_3.setStyleSheet("QLabel {\n"
" color: #FFFFFF;\n"
" }\n"
"")
def _scan():
"""Scan a new item using webcam."""
from fridgeai import camera, ai
while True:
print("Scanning...")
label = ai.predict(camera.get_frames((32, 32), count=5, interval=5))
print("Scan result: {}".format(label))
option = input("[s]ave, [r]escan or [c]ancel (default 'r'): ")
if option == "s":
# add to database
break
elif option == "c":
break
def startLearn(self, Form):
import shutil
frames = camera.get_frames(shape=(32, 32), count=100, interval=2)
label = self.Item.text()
with open(os.path.join('data', 'expiry.json'), 'r') as file:
expiry_dates = json.load(file)
with open(os.path.join('data', 'expiry.json'), 'w') as file:
expiry_dates[label] = self.Lifespan.text()
json.dump(expiry_dates, file)
os.mkdir(label)
for i, frame in enumerate(frames):
cv2.imwrite(os.path.join(label, 'frame{}.jpg'.format(i)), frame)
training.send_training_data(label)
shutil.rmtree(label)
Form.hide()
def takeSnap(self):
camera.get_frames(shape=(32, 32), count=5, interval=5)
def _snap():
"""Do a dummy scan."""
from fridgeai import camera
camera.get_frames(shape=(32, 32), count=5, interval=5)