bullseye wip

Author: previ

cnn_classifier.py

@@ -23,7 +23,15 @@
 import logging
 
 import numpy as np
-from tensorflow.lite.python.interpreter import Interpreter
+try:
+    from tensorflow.lite.python.interpreter import Interpreter
+except:
+    logging.warning("tensorflow not available (for inference)")
+try:
+    from tflite_runtime.interpreter import Interpreter
+except:
+    logging.warning("tflite not available")
+
 import cv2
 
 logger = logging.getLogger(__name__)

cnn_manager.py

@@ -26,7 +26,11 @@
 import json
 import threading
 
-from cnn_train import CNNTrainer
+try:
+    from cnn_train import CNNTrainer
+except:
+    logging.warning("tensorflow not available (for training)")
+
 from cnn_classifier import CNNClassifier
 
 MODEL_PATH = "./cnn_models"
@@ -117,6 +121,7 @@ def load_model(self, model_name):
             return CNNClassifier(model_file=MODEL_PATH + "/" + model_name + ".tflite",
                                  label_file=MODEL_PATH + "/" + model_name + ".txt")
         return None
+
     class TrainThread(threading.Thread):
 
         def __init__(self, manager, model_name, architecture, image_tags, photos_metadata, training_steps, learning_rate):

coderbot.cfg

@@ -1 +1 @@
-{"move_power_angle_3": "60", "cnn_default_model": "generic_fast_low", "prog_maxblocks": "-1", "camera_jpeg_quality": "5", "show_page_control": "true", "camera_framerate": "30", "prog_scrollbars": "true", "move_fw_speed": "100", "prog_level": "adv", "move_motor_trim": "1", "move_motor_mode": "dc", "cv_image_factor": "2", "move_power_angle_1": "45", "camera_path_object_size_min": "4000", "button_func": "none", "camera_color_object_size_min": "4000", "camera_jpeg_bitrate": "1000000", "move_fw_elapse": "1", "show_control_move_commands": "true", "camera_color_object_size_max": "160000", "show_page_prefs": "true", "camera_exposure_mode": "auto", "ctrl_tr_elapse": "-1", "show_page_program": "true", "move_tr_elapse": "0.5", "camera_path_object_size_max": "160000", "sound_shutter": "$shutter.mp3", "ctrl_fw_elapse": "-1", "sound_stop": "$shutdown.mp3", "ctrl_tr_speed": "80", "ctrl_fw_speed": "100", "move_tr_speed": "85", "move_power_angle_2": "60", "ctrl_hud_image": "", "load_at_start": "", "sound_start": "$startup.mp3", "encoder": "True", "audio_volume_level": "100", "wifi_mode": "ap", "wifi_ssid": "coderbot", "wifi_psk": "coderbot", "packages_installed": ""}
+{"move_power_angle_3": "60", "cnn_default_model": "generic_fast_low", "prog_maxblocks": "-1", "camera_jpeg_quality": "5", "show_page_control": "true", "camera_framerate": "30", "prog_scrollbars": "true", "move_fw_speed": "100", "prog_level": "adv", "move_motor_trim": "1", "move_motor_mode": "dc", "cv_image_factor": "2", "move_power_angle_1": "45", "camera_path_object_size_min": "4000", "button_func": "none", "camera_color_object_size_min": "4000", "camera_jpeg_bitrate": "1000000", "move_fw_elapse": "1", "show_control_move_commands": "true", "camera_color_object_size_max": "160000", "show_page_prefs": "true", "camera_exposure_mode": "auto", "ctrl_tr_elapse": "-1", "show_page_program": "true", "move_tr_elapse": "0.5", "camera_path_object_size_max": "160000", "sound_shutter": "$shutter.wav", "ctrl_fw_elapse": "-1", "sound_stop": "$shutdown.wav", "ctrl_tr_speed": "80", "ctrl_fw_speed": "100", "move_tr_speed": "85", "move_power_angle_2": "60", "ctrl_hud_image": "", "load_at_start": "", "sound_start": "$startup.wav", "encoder": "True", "audio_volume_level": "100", "wifi_mode": "ap", "wifi_ssid": "coderbot", "wifi_psk": "coderbot", "packages_installed": ""}