administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-12-06 21:44:13 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Improve flow handling for classification state

Browse Source
This commit is contained in:
Nicolas Mowen 2025-11-10 06:13:14 -07:00
parent 55dcbc6371
commit 8aad89a83a
1 changed files with 34 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
web/src/components/classification/wizard/Step3ChooseExamples.tsx
View File

@ -165,18 +165,15 @@ export default function Step3ChooseExamples({
const isLastClass = currentClassIndex === allClasses.length - 1; const isLastClass = currentClassIndex === allClasses.length - 1;
if (isLastClass) { if (isLastClass) {
// Assign remaining unclassified images // For object models, assign remaining unclassified images to "none"
unknownImages.slice(0, 24).forEach((imageName) => { // For state models, this should never happen since we require all images to be classified
if (!newClassifications[imageName]) { if (step1Data.modelType !== "state") {
// For state models with 2 classes, assign to the last class unknownImages.slice(0, 24).forEach((imageName) => {
// For object models, assign to "none" if (!newClassifications[imageName]) {
if (step1Data.modelType === "state" && allClasses.length === 2) {
newClassifications[imageName] = allClasses[allClasses.length - 1];
} else {
newClassifications[imageName] = "none"; newClassifications[imageName] = "none";
} }
} });
}); }
// All done, trigger training immediately // All done, trigger training immediately
setImageClassifications(newClassifications); setImageClassifications(newClassifications);
@ -316,8 +313,15 @@ export default function Step3ChooseExamples({
return images; return images;
} }
return images.filter((img) => !imageClassifications[img]); // If we're viewing a previous class (going back), show images for that class
}, [unknownImages, imageClassifications]); // Otherwise show only unclassified images
const currentClassInView = allClasses[currentClassIndex];
return images.filter((img) => {
const imgClass = imageClassifications[img];
// Show if: unclassified OR classified with current class we're viewing
return !imgClass || imgClass === currentClassInView;
});
}, [unknownImages, imageClassifications, allClasses, currentClassIndex]);
const allImagesClassified = useMemo(() => { const allImagesClassified = useMemo(() => {
return unclassifiedImages.length === 0; return unclassifiedImages.length === 0;
@ -326,15 +330,26 @@ export default function Step3ChooseExamples({
// For state models on the last class, require all images to be classified // For state models on the last class, require all images to be classified
const isLastClass = currentClassIndex === allClasses.length - 1; const isLastClass = currentClassIndex === allClasses.length - 1;
const canProceed = useMemo(() => { const canProceed = useMemo(() => {
if ( if (step1Data.modelType === "state" && isLastClass) {
step1Data.modelType === "state" && // Check if all 24 images will be classified after current selections are applied
isLastClass && const totalImages = unknownImages.slice(0, 24).length;
!allImagesClassified
) { // Count images that will be classified (either already classified or currently selected)
return false; const allImages = unknownImages.slice(0, 24);
const willBeClassified = allImages.filter((img) => {
return imageClassifications[img] || selectedImages.has(img);
}).length;
return willBeClassified >= totalImages;
} }
return true; return true;
}, [step1Data.modelType, isLastClass, allImagesClassified]); }, [
step1Data.modelType,
isLastClass,
unknownImages,
imageClassifications,
selectedImages,
]);
const handleBack = useCallback(() => { const handleBack = useCallback(() => {
if (currentClassIndex > 0) { if (currentClassIndex > 0) {