Cleanup classification

This commit is contained in:
Nicolas Mowen 2025-10-22 06:44:35 -06:00
parent 10e9080093
commit 7e738c6538

View File

@ -632,12 +632,11 @@ function DatasetGrid({
return (
<div
ref={contentRef}
className="scrollbar-container flex flex-wrap gap-2 overflow-y-auto p-2"
className="scrollbar-container grid grid-cols-2 gap-2 overflow-y-scroll p-1 md:grid-cols-4 xl:grid-cols-8 2xl:grid-cols-10 3xl:grid-cols-12"
>
{classData.map((image) => (
<div key={image} className="aspect-square w-full">
<ClassificationCard
key={image}
imgClassName="size-auto"
data={{
filename: image,
filepath: `clips/${modelName}/dataset/${categoryName}/${image}`,
@ -662,6 +661,7 @@ function DatasetGrid({
</TooltipContent>
</Tooltip>
</ClassificationCard>
</div>
))}
</div>
);
@ -791,14 +791,12 @@ function StateTrainGrid({
<div
ref={contentRef}
className={cn(
"scrollbar-container flex flex-wrap gap-3 overflow-y-auto p-2",
isMobile && "justify-center",
"scrollbar-container grid grid-cols-2 gap-3 overflow-y-scroll p-1 sm:grid-cols-3 md:grid-cols-4 lg:grid-cols-6 xl:grid-cols-8 2xl:grid-cols-10 3xl:grid-cols-12",
)}
>
{trainData?.map((data) => (
<div key={data.filename} className="aspect-square w-full">
<ClassificationCard
key={data.filename}
imgClassName="size-auto"
data={data}
threshold={threshold}
selected={selectedImages.includes(data.filename)}
@ -815,6 +813,7 @@ function StateTrainGrid({
<TbCategoryPlus className="size-7 cursor-pointer p-1 text-gray-200 hover:rounded-full hover:bg-primary-foreground" />
</ClassificationSelectionDialog>
</ClassificationCard>
</div>
))}
</div>
);
@ -934,12 +933,8 @@ function ObjectTrainGrid({
{Object.entries(groups).map(([key, group]) => {
const event = events?.find((ev) => ev.id == key);
return (
<div
key={key}
className={cn(isMobile && "aspect-square size-full")}
>
<div key={key} className="aspect-square w-full">
<GroupedClassificationCard
key={key}
group={group}
event={event}
threshold={threshold}