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

View File

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