Add batch classification assignment to training view

Co-authored-by: Teagan42 <2989925+Teagan42@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot] 2026-02-19 08:12:50 +00:00
parent 220f3e1e7e
commit e516bf5fda
2 changed files with 9 additions and 2 deletions

View File

@ -13,7 +13,8 @@
"trainModel": "Train Model",
"addClassification": "Add Classification",
"deleteModels": "Delete Models",
"editModel": "Edit Model"
"editModel": "Edit Model",
"categorizeImages": "Classify Images"
},
"tooltip": {
"trainingInProgress": "Model is currently training",
@ -28,6 +29,7 @@
"deletedModel_one": "Successfully deleted {{count}} model",
"deletedModel_other": "Successfully deleted {{count}} models",
"categorizedImage": "Successfully Classified Image",
"batchCategorized": "Successfully classified {{count}} images",
"trainedModel": "Successfully trained model.",
"trainingModel": "Successfully started model training.",
"updatedModel": "Successfully updated model configuration",
@ -38,10 +40,14 @@
"deleteCategoryFailed": "Failed to delete class: {{errorMessage}}",
"deleteModelFailed": "Failed to delete model: {{errorMessage}}",
"categorizeFailed": "Failed to categorize image: {{errorMessage}}",
"batchCategorizeFailed": "Failed to classify {{count}} images",
"trainingFailed": "Model training failed. Check Frigate logs for details.",
"trainingFailedToStart": "Failed to start model training: {{errorMessage}}",
"updateModelFailed": "Failed to update model: {{errorMessage}}",
"renameCategoryFailed": "Failed to rename class: {{errorMessage}}"
},
"warning": {
"partialBatchCategorized": "Classified {{success}} of {{total}} images successfully."
}
},
"deleteCategory": {

View File

@ -463,7 +463,8 @@ export default function ModelTrainingView({ model }: ModelTrainingViewProps) {
classes={classes}
modelName={model.name}
image={selectedImages[0]}
onRefresh={(category: string) => {
onRefresh={refreshAll}
onCategorize={(category) => {
// Batch categorize all selected images
let successCount = 0;
let failCount = 0;