nik/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-02 06:23:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bugs

Browse Source
This commit is contained in:
Nicolas Mowen 2025-10-22 20:48:47 -06:00
parent 885eef6026
commit 9160c5168e
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
frigate/util/classification.py
View File

@ -75,6 +75,7 @@ class ClassificationTrainingProcess(FrigateProcess):
logger.info(f"Kicking off classification training for {self.model_name}.")
dataset_dir = os.path.join(CLIPS_DIR, self.model_name, "dataset")
model_dir = os.path.join(MODEL_CACHE_DIR, self.model_name)
os.makedirs(model_dir, exist_ok=True)
num_classes = len(
[
d

10
web/src/views/classification/ModelSelectionView.tsx
View File

@ -67,7 +67,15 @@ export default function ModelSelectionView({
}
if (classificationConfigs.length == 0) {
return <NoModelsView onCreateModel={() => setNewModel(true)} />;
return (
<>
<ClassificationModelWizardDialog
open={newModel}
onClose={() => setNewModel(false)}
/>
<NoModelsView onCreateModel={() => setNewModel(true)} />;
</>
);
}
return (