Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-20 13:54:36 +01:00
Code Issues Projects Releases Wiki Activity

Classification Model UI (#18571)

Browse Source 
* Setup basic training structure

* Build out route

* Handle model configs

* Add image fetch APIs

* Implement model training screen with dataset selection

* Implement viewing of training images

* Adjust directories

* Implement viewing of images

* Add support for deleting images

* Implement full deletion

* Implement classification model training

* Improve naming

* More renaming

* Improve layout

* Reduce logging

* Cleanup
This commit is contained in:
Nicolas Mowen
2025-06-04 17:09:55 -06:00
committed by Blake Blackshear
parent ac7fb29b32
commit 1c75ff59f1
11 changed files with 1156 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -42,7 +42,7 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
self.model_config = model_config
self.requestor = requestor
self.model_dir = os.path.join(MODEL_CACHE_DIR, self.model_config.name)
self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name)
self.train_dir = os.path.join(CLIPS_DIR, self.model_config.name, "train")
self.interpreter: Interpreter = None
self.tensor_input_details: dict[str, Any] = None
self.tensor_output_details: dict[str, Any] = None