Improve object classification (#18908)

* Ui improvements

* Improve image cropping and model saving

* Improve naming

* Add logs for training

* Improve model labeling

* Don't set sub label for none object classification

* Cleanup
This commit is contained in:
Nicolas Mowen
2025-08-16 10:20:33 -05:00
committed by Blake Blackshear
parent ceeb6543f5
commit 13fb7bc260
4 changed files with 35 additions and 26 deletions
@@ -82,7 +82,7 @@ export default function ClassificationSelectionDialog({
);
// control
const [newFace, setNewFace] = useState(false);
const [newClass, setNewClass] = useState(false);
// components
const Selector = isDesktop ? DropdownMenu : Drawer;
@@ -98,10 +98,10 @@ export default function ClassificationSelectionDialog({
return (
<div className={className ?? ""}>
{newFace && (
{newClass && (
<TextEntryDialog
open={true}
setOpen={setNewFace}
setOpen={setNewClass}
title={t("createCategory.new")}
onSave={(newCat) => onCategorizeImage(newCat)}
/>
@@ -130,7 +130,7 @@ export default function ClassificationSelectionDialog({
>
<SelectorItem
className="flex cursor-pointer gap-2 smart-capitalize"
onClick={() => setNewFace(true)}
onClick={() => setNewClass(true)}
>
<LuPlus />
{t("createCategory.new")}
@@ -142,7 +142,7 @@ export default function ClassificationSelectionDialog({
onClick={() => onCategorizeImage(category)}
>
<MdCategory />
{category}
{category.replaceAll("_", " ")}
</SelectorItem>
))}
</div>