diff --git a/docker/main/Dockerfile b/docker/main/Dockerfile index f58e3d5a9..959ff59d4 100644 --- a/docker/main/Dockerfile +++ b/docker/main/Dockerfile @@ -227,6 +227,9 @@ ENV OPENCV_FFMPEG_LOGLEVEL=8 # Set HailoRT to disable logging ENV HAILORT_LOGGER_PATH=NONE +# TensorFlow error only +ENV TF_CPP_MIN_LOG_LEVEL=3 + ENV PATH="/usr/local/go2rtc/bin:/usr/local/tempio/bin:/usr/local/nginx/sbin:${PATH}" # Install dependencies diff --git a/frigate/util/classification.py b/frigate/util/classification.py index e281a0535..55a204616 100644 --- a/frigate/util/classification.py +++ b/frigate/util/classification.py @@ -87,7 +87,7 @@ def train_classification_model(model_dir: str) -> bool: f.write(f"{class_name}\n") # train the model - model.fit(train_gen, epochs=EPOCHS) + model.fit(train_gen, epochs=EPOCHS, verbose=0) # convert model to tflite converter = tf.lite.TFLiteConverter.from_keras_model(model)