def main():
try:
open("settings.ini", "r").close()
first_time = False
except FileNotFoundError:
first_time = True
update_exit = check_version(version)
if update_exit:
return
setup_completed = not first_time or first_time_setup()
if setup_completed:
generate_default_bases()
log_settings(version)
settings = ConfigParser()
settings.read("settings.ini")
editor = WaypointEditor(settings)
gui = GUI(editor, version)
try:
gui.run()
except Exception:
gui.close()
raise
def main(project_parameters):
result = None
if project_parameters.mode == 'train':
result = train(project_parameters=project_parameters)
elif project_parameters.mode == 'evaluate':
if project_parameters.predefined_dataset is not None:
print('temporarily does not support predefined dataset.')
else:
evaluate(project_parameters=project_parameters)
elif project_parameters.mode == 'predict':
if project_parameters.use_gui:
gui = GUI(project_parameters=project_parameters)
gui.run()
else:
result = Predict(project_parameters=project_parameters)(
data_path=project_parameters.data_path)
print(('{},' * project_parameters.num_classes).format(
*project_parameters.classes)[:-1])
print(result)
elif project_parameters.mode == 'tune':
result = tune(project_parameters=project_parameters)
return result
class Main(object):
def __init__(self):
self.gui = GUI()
self.camera = Camera()
self.camera.startVideo()
self.detectCanny = False
self.detectFace = False
self.detectEye = False
def funcDetectCanny(self):
if self.detectCanny:
self.detectCanny = False
self.gui.btnCan.configure(text="Turn on canny detection")
else:
self.detectCanny = True
self.gui.btnCan.configure(text="Turn off canny detection")
self.camera.closeWindow()
def funcDetectFace(self):
if self.detectFace:
self.detectFace = False
self.gui.btnFace.configure(text="Turn on face detection")
else:
self.detectFace = True
self.gui.btnFace.configure(text="Turn off face detection")
self.camera.closeWindow()
def funcDetectEye(self):
if self.detectEye:
self.detectEye = False
self.gui.btnEye.configure(text="Turn on face detectoe")
else:
self.detectEye = True
self.gui.btnEye.config(text="Turn off face detector")
self.camera.closeWindow()
def video_stream(self):
ret, frame = self.camera.readVideo(color=True)
if ret:
if self.detectCanny:
can = self.camera.convertToCanny(frame)
self.camera.show("Canny Result: ", can)
elif self.detectFace:
rect = self.camera.detectFace()
self.camera.show("Face", rect)
elif self.detectEye:
rect = self.camera.detectEye()
self.camera.show("Eye", rect)
else:
self.camera.closeWindow()
photo = self.camera.convertToTk(
self.camera.convertFromArray(frame))
self.gui.set_frame(photo)
self.gui.update()
self.gui.after(15, self.video_stream)
def run(self):
self.gui.createButton(command=self.camera.snapshot)
self.gui.createButtonCanny(command=self.funcDetectCanny)
self.gui.createButtonFace(command=self.funcDetectFace)
self.gui.createButtonEye(command=self.funcDetectEye)
self.video_stream()
self.gui.run()
from src.gui import GUI
if __name__ == "__main__":
gui = GUI()
gui.run()
