administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-01-29 23:44:55 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Always mark model as untrained when a classname is changed

Browse Source
This commit is contained in:
Nicolas Mowen 2025-12-19 07:54:17 -07:00
parent 73eceaf242
commit 853723795c
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/api/classification.py
View File

@ -40,6 +40,7 @@ from frigate.util.classification import (
collect_state_classification_examples, collect_state_classification_examples,
get_dataset_image_count, get_dataset_image_count,
read_training_metadata, read_training_metadata,
write_training_metadata,
) )
from frigate.util.file import get_event_snapshot from frigate.util.file import get_event_snapshot
@ -842,6 +843,12 @@ def rename_classification_category(
try: try:
os.rename(old_folder, new_folder) os.rename(old_folder, new_folder)
# Mark dataset as ready to train by resetting training metadata
# This ensures the dataset is marked as changed after renaming
sanitized_name = sanitize_filename(name)
write_training_metadata(sanitized_name, 0)
return JSONResponse( return JSONResponse(
content=( content=(
{ {