administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 19:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Reduce logging

Browse Source
This commit is contained in:
Nicolas Mowen 2025-06-04 13:37:17 -06:00
parent 11203398eb
commit c7e7befb7b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/util/classification.py
View File

@ -1,5 +1,6 @@
"""Util for classification models."""
import logging
import os
import cv2
@ -49,6 +50,8 @@ def train_classification_model(model_name: str) -> bool:
]
)
tf.get_logger().setLevel(logging.ERROR)
# Start with imagenet base model with 35% of channels in each layer
base_model = MobileNetV2(
input_shape=(224, 224, 3),