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

Fix train dir

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-29 12:36:22 -06:00
parent 4a5e6d3f97
commit 93ccd7f311
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/data_processing/real_time/custom_classification.py
View File

@ -15,7 +15,7 @@ from frigate.comms.event_metadata_updater import (
from frigate.comms.inter_process import InterProcessRequestor from frigate.comms.inter_process import InterProcessRequestor
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.config.classification import CustomClassificationConfig from frigate.config.classification import CustomClassificationConfig
from frigate.const import MODEL_CACHE_DIR from frigate.const import CLIPS_DIR, MODEL_CACHE_DIR
from frigate.util.builtin import load_labels from frigate.util.builtin import load_labels
from frigate.util.object import box_overlaps, calculate_region from frigate.util.object import box_overlaps, calculate_region
@ -42,7 +42,7 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
self.model_config = model_config self.model_config = model_config
self.requestor = requestor self.requestor = requestor
self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name) self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name)
self.train_dir = os.path.join(self.model_dir, "train") self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name)
self.interpreter: Interpreter = None self.interpreter: Interpreter = None
self.tensor_input_details: dict[str, Any] = None self.tensor_input_details: dict[str, Any] = None
self.tensor_output_details: dict[str, Any] = None self.tensor_output_details: dict[str, Any] = None