Model manager for DL models to preannotate images (#289)
Andrey Zhavoronkov
7 years ago
committed by
Nikita Manovich
parent
0dcf211d0c
commit
746cffb476
@ -0,0 +1,39 @@
|
|||||||
|
# Generated by Django 2.1.3 on 2019-01-24 14:05
|
||||||
|
|
||||||
|
import cvat.apps.auto_annotation.models
|
||||||
|
from django.conf import settings
|
||||||
|
import django.core.files.storage
|
||||||
|
from django.db import migrations, models
|
||||||
|
import django.db.models.deletion
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
initial = True
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='AnnotationModel',
|
||||||
|
fields=[
|
||||||
|
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('name', cvat.apps.auto_annotation.models.SafeCharField(max_length=256)),
|
||||||
|
('created_date', models.DateTimeField(auto_now_add=True)),
|
||||||
|
('updated_date', models.DateTimeField(auto_now_add=True)),
|
||||||
|
('model_file', models.FileField(storage=django.core.files.storage.FileSystemStorage(), upload_to=cvat.apps.auto_annotation.models.upload_path_handler)),
|
||||||
|
('weights_file', models.FileField(storage=django.core.files.storage.FileSystemStorage(), upload_to=cvat.apps.auto_annotation.models.upload_path_handler)),
|
||||||
|
('labelmap_file', models.FileField(storage=django.core.files.storage.FileSystemStorage(), upload_to=cvat.apps.auto_annotation.models.upload_path_handler)),
|
||||||
|
('interpretation_file', models.FileField(storage=django.core.files.storage.FileSystemStorage(), upload_to=cvat.apps.auto_annotation.models.upload_path_handler)),
|
||||||
|
('shared', models.BooleanField(default=False)),
|
||||||
|
('primary', models.BooleanField(default=False)),
|
||||||
|
('framework', models.CharField(default=cvat.apps.auto_annotation.models.FrameworkChoice('openvino'), max_length=32)),
|
||||||
|
('owner', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL)),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'default_permissions': (),
|
||||||
|
},
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,143 @@
|
|||||||
|
# Copyright (C) 2018 Intel Corporation
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
|
import django_rq
|
||||||
|
import os
|
||||||
|
import rq
|
||||||
|
import shutil
|
||||||
|
|
||||||
|
from django.db import transaction
|
||||||
|
from django.utils import timezone
|
||||||
|
from django.conf import settings
|
||||||
|
|
||||||
|
from .models import AnnotationModel, FrameworkChoice
|
||||||
|
|
||||||
|
def _remove_old_file(model_file_field):
|
||||||
|
if model_file_field and os.path.exists(model_file_field.name):
|
||||||
|
os.remove(model_file_field.name)
|
||||||
|
|
||||||
|
@transaction.atomic
|
||||||
|
def _update_dl_model_thread(dl_model_id, model_file, weights_file, labelmap_file, interpretation_file, run_tests):
|
||||||
|
def _get_file_content(filename):
|
||||||
|
return os.path.basename(filename), open(filename, "rb")
|
||||||
|
|
||||||
|
job = rq.get_current_job()
|
||||||
|
job.meta["progress"] = "Saving data"
|
||||||
|
job.save_meta()
|
||||||
|
|
||||||
|
dl_model = AnnotationModel.objects.select_for_update().get(pk=dl_model_id)
|
||||||
|
|
||||||
|
#save files in case of files should be uploaded from share
|
||||||
|
if model_file:
|
||||||
|
_remove_old_file(dl_model.model_file)
|
||||||
|
dl_model.model_file.save(*_get_file_content(model_file))
|
||||||
|
if weights_file:
|
||||||
|
_remove_old_file(dl_model.weights_file)
|
||||||
|
dl_model.weights_file.save(*_get_file_content(weights_file))
|
||||||
|
if labelmap_file:
|
||||||
|
_remove_old_file(dl_model.labelmap_file)
|
||||||
|
dl_model.labelmap_file.save(*_get_file_content(labelmap_file))
|
||||||
|
if interpretation_file:
|
||||||
|
_remove_old_file(dl_model.interpretation_file)
|
||||||
|
dl_model.interpretation_file.save(*_get_file_content(interpretation_file))
|
||||||
|
|
||||||
|
if run_tests:
|
||||||
|
#only for testing
|
||||||
|
import time
|
||||||
|
time.sleep(3)
|
||||||
|
job.meta["progress"] = "Test started"
|
||||||
|
job.save_meta()
|
||||||
|
time.sleep(5)
|
||||||
|
job.meta["progress"] = "Test finished"
|
||||||
|
|
||||||
|
@transaction.atomic
|
||||||
|
def update_model(dl_model_id, name, model_file, weights_file, labelmap_file, interpretation_file, storage, is_shared):
|
||||||
|
|
||||||
|
def get_abs_path(share_path):
|
||||||
|
if not share_path:
|
||||||
|
return share_path
|
||||||
|
share_root = settings.SHARE_ROOT
|
||||||
|
relpath = os.path.normpath(share_path).lstrip('/')
|
||||||
|
if '..' in relpath.split(os.path.sep):
|
||||||
|
raise Exception('Permission denied')
|
||||||
|
abspath = os.path.abspath(os.path.join(share_root, relpath))
|
||||||
|
if os.path.commonprefix([share_root, abspath]) != share_root:
|
||||||
|
raise Exception('Bad file path on share: ' + abspath)
|
||||||
|
return abspath
|
||||||
|
|
||||||
|
dl_model = AnnotationModel.objects.select_for_update().get(pk=dl_model_id)
|
||||||
|
|
||||||
|
if name:
|
||||||
|
dl_model.name = name
|
||||||
|
|
||||||
|
if is_shared != None:
|
||||||
|
dl_model.shared = is_shared
|
||||||
|
|
||||||
|
run_tests = bool(model_file or weights_file or labelmap_file or interpretation_file)
|
||||||
|
if storage != "local":
|
||||||
|
model_file = get_abs_path(model_file)
|
||||||
|
weights_file = get_abs_path(weights_file)
|
||||||
|
labelmap_file = get_abs_path(labelmap_file)
|
||||||
|
interpretation_file = get_abs_path(interpretation_file)
|
||||||
|
else:
|
||||||
|
if model_file:
|
||||||
|
_remove_old_file(dl_model.model_file)
|
||||||
|
dl_model.model_file = model_file
|
||||||
|
model_file = None
|
||||||
|
if weights_file:
|
||||||
|
_remove_old_file(dl_model.weights_file)
|
||||||
|
dl_model.weights_file = weights_file
|
||||||
|
weights_file = None
|
||||||
|
if labelmap_file:
|
||||||
|
_remove_old_file(dl_model.labelmap_file)
|
||||||
|
dl_model.labelmap_file = labelmap_file
|
||||||
|
labelmap_file = None
|
||||||
|
if interpretation_file:
|
||||||
|
_remove_old_file(dl_model.interpretation_file)
|
||||||
|
dl_model.interpretation_file = interpretation_file
|
||||||
|
interpretation_file = None
|
||||||
|
|
||||||
|
dl_model.updated_date = timezone.now()
|
||||||
|
dl_model.save()
|
||||||
|
|
||||||
|
rq_id = "auto_annotation.create.{}".format(dl_model_id)
|
||||||
|
queue = django_rq.get_queue('default')
|
||||||
|
queue.enqueue_call(
|
||||||
|
func = _update_dl_model_thread,
|
||||||
|
args = (dl_model_id,
|
||||||
|
model_file,
|
||||||
|
weights_file,
|
||||||
|
labelmap_file,
|
||||||
|
interpretation_file,
|
||||||
|
run_tests,
|
||||||
|
),
|
||||||
|
job_id = rq_id
|
||||||
|
)
|
||||||
|
|
||||||
|
return rq_id
|
||||||
|
|
||||||
|
def create_empty(owner, framework=FrameworkChoice.OPENVINO):
|
||||||
|
db_model = AnnotationModel(
|
||||||
|
owner=owner,
|
||||||
|
)
|
||||||
|
db_model.save()
|
||||||
|
|
||||||
|
model_path = db_model.get_dirname()
|
||||||
|
if os.path.isdir(model_path):
|
||||||
|
shutil.rmtree(model_path)
|
||||||
|
os.mkdir(model_path)
|
||||||
|
|
||||||
|
return db_model.id
|
||||||
|
|
||||||
|
@transaction.atomic
|
||||||
|
def delete(dl_model_id):
|
||||||
|
dl_model = AnnotationModel.objects.select_for_update().get(pk=dl_model_id)
|
||||||
|
if dl_model:
|
||||||
|
if dl_model.primary:
|
||||||
|
raise Exception("Can not delete primary model {}".format(dl_model_id))
|
||||||
|
|
||||||
|
dl_model.delete()
|
||||||
|
shutil.rmtree(dl_model.get_dirname(), ignore_errors=True)
|
||||||
|
else:
|
||||||
|
raise Exception("Requested DL model {} doesn't exist".format(dl_model_id))
|
||||||
@ -0,0 +1,29 @@
|
|||||||
|
# Copyright (C) 2018 Intel Corporation
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
|
import rules
|
||||||
|
|
||||||
|
from cvat.apps.authentication.auth import has_admin_role, has_user_role
|
||||||
|
|
||||||
|
@rules.predicate
|
||||||
|
def is_model_owner(db_user, db_dl_model):
|
||||||
|
return db_dl_model.owner == db_user
|
||||||
|
|
||||||
|
@rules.predicate
|
||||||
|
def is_shared_model(_, db_dl_model):
|
||||||
|
return db_dl_model.shared
|
||||||
|
|
||||||
|
@rules.predicate
|
||||||
|
def is_primary_model(_, db_dl_model):
|
||||||
|
return db_dl_model.primary
|
||||||
|
|
||||||
|
def setup_permissions():
|
||||||
|
rules.add_perm('auto_annotation.model.create', has_admin_role | has_user_role)
|
||||||
|
|
||||||
|
rules.add_perm('auto_annotation.model.update', (has_admin_role | is_model_owner) & ~is_primary_model)
|
||||||
|
|
||||||
|
rules.add_perm('auto_annotation.model.delete', (has_admin_role | is_model_owner) & ~is_primary_model)
|
||||||
|
|
||||||
|
rules.add_perm('auto_annotation.model.access', has_admin_role | is_model_owner |
|
||||||
|
is_shared_model | is_primary_model)
|
||||||
@ -1,188 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (C) 2018 Intel Corporation
|
|
||||||
*
|
|
||||||
* SPDX-License-Identifier: MIT
|
|
||||||
*/
|
|
||||||
|
|
||||||
/* global
|
|
||||||
showMessage:false
|
|
||||||
*/
|
|
||||||
|
|
||||||
"use strict";
|
|
||||||
|
|
||||||
window.cvat = window.cvat || {};
|
|
||||||
window.cvat.dashboard = window.cvat.dashboard || {};
|
|
||||||
window.cvat.dashboard.uiCallbacks = window.cvat.dashboard.uiCallbacks || [];
|
|
||||||
window.cvat.dashboard.uiCallbacks.push(function(newElements) {
|
|
||||||
let tids = [];
|
|
||||||
for (let el of newElements) {
|
|
||||||
tids.push(el.id.split("_")[1]);
|
|
||||||
}
|
|
||||||
|
|
||||||
$.ajax({
|
|
||||||
type: "POST",
|
|
||||||
url: "/auto_annotation/meta/get",
|
|
||||||
data: JSON.stringify(tids),
|
|
||||||
contentType: "application/json; charset=utf-8",
|
|
||||||
success: (data) => {
|
|
||||||
newElements.each(function() {
|
|
||||||
let elem = $(this);
|
|
||||||
let tid = +elem.attr("id").split("_")[1];
|
|
||||||
|
|
||||||
const autoAnnoButton = $("<button> Run auto annotation </button>").addClass("regular dashboardButtonUI dashboardAutoAnno");
|
|
||||||
autoAnnoButton.appendTo(elem.find("div.dashboardButtonsUI")[0]);
|
|
||||||
|
|
||||||
if (tid in data && data[tid].active) {
|
|
||||||
autoAnnoButton.text("Cancel auto annotation");
|
|
||||||
autoAnnoButton.addClass("autoAnnotationProcess");
|
|
||||||
window.cvat.autoAnnotation.checkAutoAnnotationRequest(tid, autoAnnoButton);
|
|
||||||
}
|
|
||||||
|
|
||||||
autoAnnoButton.on("click", () => {
|
|
||||||
if (autoAnnoButton.hasClass("autoAnnotationProcess")) {
|
|
||||||
$.post(`/auto_annotation/cancel/task/${tid}`).fail( (data) => {
|
|
||||||
let message = `Error during cancel auto annotation request. Code: ${data.status}. Message: ${data.responseText || data.statusText}`;
|
|
||||||
showMessage(message);
|
|
||||||
throw Error(message);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
let dialogWindow = $(`#${window.cvat.autoAnnotation.modalWindowId}`);
|
|
||||||
dialogWindow.attr("current_tid", tid);
|
|
||||||
dialogWindow.removeClass("hidden");
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
},
|
|
||||||
error: (data) => {
|
|
||||||
let message = `Can not get auto annotation meta info. Code: ${data.status}. Message: ${data.responseText || data.statusText}`;
|
|
||||||
window.cvat.autoAnnotation.badResponse(message);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
window.cvat.autoAnnotation = {
|
|
||||||
modalWindowId: "autoAnnotationWindow",
|
|
||||||
autoAnnoFromId: "autoAnnotationForm",
|
|
||||||
autoAnnoModelFieldId: "autoAnnotationModelField",
|
|
||||||
autoAnnoWeightsFieldId: "autoAnnotationWeightsField",
|
|
||||||
autoAnnoConfigFieldId: "autoAnnotationConfigField",
|
|
||||||
autoAnnoConvertFieldId: "autoAnnotationConvertField",
|
|
||||||
autoAnnoCloseButtonId: "autoAnnoCloseButton",
|
|
||||||
autoAnnoSubmitButtonId: "autoAnnoSubmitButton",
|
|
||||||
|
|
||||||
checkAutoAnnotationRequest: (tid, autoAnnoButton) => {
|
|
||||||
function timeoutCallback() {
|
|
||||||
$.get(`/auto_annotation/check/task/${tid}`).done((data) => {
|
|
||||||
if (data.status === "started" || data.status === "queued") {
|
|
||||||
let progress = Math.round(data.progress) || 0;
|
|
||||||
autoAnnoButton.text(`Cancel auto annotation (${progress}%)`);
|
|
||||||
setTimeout(timeoutCallback, 1000);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
autoAnnoButton.text("Run auto annotation");
|
|
||||||
autoAnnoButton.removeClass("autoAnnotationProcess");
|
|
||||||
}
|
|
||||||
}).fail((data) => {
|
|
||||||
let message = "Error was occurred during check annotation status. " +
|
|
||||||
`Code: ${data.status}, text: ${data.responseText || data.statusText}`;
|
|
||||||
window.cvat.autoAnnotation.badResponse(message);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
setTimeout(timeoutCallback, 1000);
|
|
||||||
},
|
|
||||||
|
|
||||||
badResponse(message) {
|
|
||||||
showMessage(message);
|
|
||||||
throw Error(message);
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
function submitButtonOnClick() {
|
|
||||||
const annoWindow = $(`#${window.cvat.autoAnnotation.modalWindowId}`);
|
|
||||||
const tid = annoWindow.attr("current_tid");
|
|
||||||
const modelInput = $(`#${window.cvat.autoAnnotation.autoAnnoModelFieldId}`);
|
|
||||||
const weightsInput = $(`#${window.cvat.autoAnnotation.autoAnnoWeightsFieldId}`);
|
|
||||||
const configInput = $(`#${window.cvat.autoAnnotation.autoAnnoConfigFieldId}`);
|
|
||||||
const convFileInput = $(`#${window.cvat.autoAnnotation.autoAnnoConvertFieldId}`);
|
|
||||||
|
|
||||||
const modelFile = modelInput.prop("files")[0];
|
|
||||||
const weightsFile = weightsInput.prop("files")[0];
|
|
||||||
const configFile = configInput.prop("files")[0];
|
|
||||||
const convFile = convFileInput.prop("files")[0];
|
|
||||||
|
|
||||||
if (!modelFile || !weightsFile || !configFile || !convFile) {
|
|
||||||
showMessage("All files must be selected");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let taskData = new FormData();
|
|
||||||
taskData.append("model", modelFile);
|
|
||||||
taskData.append("weights", weightsFile);
|
|
||||||
taskData.append("config", configFile);
|
|
||||||
taskData.append("conv_script", convFile);
|
|
||||||
|
|
||||||
$.ajax({
|
|
||||||
url: `/auto_annotation/create/task/${tid}`,
|
|
||||||
type: "POST",
|
|
||||||
data: taskData,
|
|
||||||
contentType: false,
|
|
||||||
processData: false,
|
|
||||||
}).done(() => {
|
|
||||||
annoWindow.addClass("hidden");
|
|
||||||
const autoAnnoButton = $(`#dashboardTask_${tid} div.dashboardButtonsUI button.dashboardAutoAnno`);
|
|
||||||
autoAnnoButton.addClass("autoAnnotationProcess");
|
|
||||||
window.cvat.autoAnnotation.checkAutoAnnotationRequest(tid, autoAnnoButton);
|
|
||||||
}).fail((data) => {
|
|
||||||
let message = "Error was occurred during run annotation request. " +
|
|
||||||
`Code: ${data.status}, text: ${data.responseText || data.statusText}`;
|
|
||||||
window.cvat.autoAnnotation.badResponse(message);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
document.addEventListener("DOMContentLoaded", () => {
|
|
||||||
$(`<div id="${window.cvat.autoAnnotation.modalWindowId}" class="modal hidden">
|
|
||||||
<form id="${window.cvat.autoAnnotation.autoAnnoFromId}" class="modal-content" autocomplete="on" onsubmit="return false" style="width: 700px;">
|
|
||||||
<center>
|
|
||||||
<label class="semiBold h1"> Auto annotation setup </label>
|
|
||||||
</center>
|
|
||||||
|
|
||||||
<table style="width: 100%; text-align: left;">
|
|
||||||
<tr>
|
|
||||||
<td style="width: 25%"> <label class="regular h2"> Model </label> </td>
|
|
||||||
<td> <input id="${window.cvat.autoAnnotation.autoAnnoModelFieldId}" type="file" name="model" /> </td>
|
|
||||||
</tr>
|
|
||||||
<tr>
|
|
||||||
<td style="width: 25%"> <label class="regular h2"> Weights </label> </td>
|
|
||||||
<td> <input id="${window.cvat.autoAnnotation.autoAnnoWeightsFieldId}" type="file" name="weights" /> </td>
|
|
||||||
</tr>
|
|
||||||
<tr>
|
|
||||||
<td style="width: 25%"> <label class="regular h2"> Label map </label> </td>
|
|
||||||
<td> <input id="${window.cvat.autoAnnotation.autoAnnoConfigFieldId}" type="file" name="config" accept=".json" /> </td>
|
|
||||||
</tr>
|
|
||||||
<tr>
|
|
||||||
<td style="width: 25%"> <label class="regular h2"> Convertation script </label> </td>
|
|
||||||
<td> <input id="${window.cvat.autoAnnotation.autoAnnoConvertFieldId}" type="file" name="convert" /> </td>
|
|
||||||
</tr>
|
|
||||||
</table>
|
|
||||||
<div>
|
|
||||||
<button id="${window.cvat.autoAnnotation.autoAnnoCloseButtonId}" class="regular h2"> Close </button>
|
|
||||||
<button id="${window.cvat.autoAnnotation.autoAnnoSubmitButtonId}" class="regular h2"> Submit </button>
|
|
||||||
</div>
|
|
||||||
</form>
|
|
||||||
|
|
||||||
</div>`).appendTo("body");
|
|
||||||
|
|
||||||
const annoWindow = $(`#${window.cvat.autoAnnotation.modalWindowId}`);
|
|
||||||
const closeWindowButton = $(`#${window.cvat.autoAnnotation.autoAnnoCloseButtonId}`);
|
|
||||||
const submitButton = $(`#${window.cvat.autoAnnotation.autoAnnoSubmitButtonId}`);
|
|
||||||
|
|
||||||
|
|
||||||
closeWindowButton.on("click", () => {
|
|
||||||
annoWindow.addClass("hidden");
|
|
||||||
});
|
|
||||||
|
|
||||||
submitButton.on("click", () => {
|
|
||||||
submitButtonOnClick();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@ -0,0 +1,792 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (C) 2018 Intel Corporation
|
||||||
|
*
|
||||||
|
* SPDX-License-Identifier: MIT
|
||||||
|
*/
|
||||||
|
|
||||||
|
/* global
|
||||||
|
showMessage
|
||||||
|
showOverlay
|
||||||
|
userConfirm
|
||||||
|
*/
|
||||||
|
|
||||||
|
window.cvat = window.cvat || {};
|
||||||
|
window.cvat.dashboard = window.cvat.dashboard || {};
|
||||||
|
window.cvat.dashboard.uiCallbacks = window.cvat.dashboard.uiCallbacks || [];
|
||||||
|
|
||||||
|
const AutoAnnotationServer = {
|
||||||
|
start(modelId, taskId, data, success, error, progress, check) {
|
||||||
|
$.ajax({
|
||||||
|
url: `/auto_annotation/start/${modelId}/${taskId}`,
|
||||||
|
type: 'POST',
|
||||||
|
data: JSON.stringify(data),
|
||||||
|
contentType: 'application/json',
|
||||||
|
success: (responseData) => {
|
||||||
|
check(responseData.id, success, error, progress);
|
||||||
|
},
|
||||||
|
error: (responseData) => {
|
||||||
|
const message = `Starting request has been failed. Code: ${responseData.status}. Message: ${responseData.responseText || responseData.statusText}`;
|
||||||
|
error(message);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
update(data, success, error, progress, check, modelId) {
|
||||||
|
let url = '';
|
||||||
|
if (modelId === null) {
|
||||||
|
url = '/auto_annotation/create';
|
||||||
|
} else {
|
||||||
|
url = `/auto_annotation/update/${modelId}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
$.ajax({
|
||||||
|
url,
|
||||||
|
type: 'POST',
|
||||||
|
data,
|
||||||
|
contentType: false,
|
||||||
|
processData: false,
|
||||||
|
success: (responseData) => {
|
||||||
|
check(responseData.id, success, error, progress);
|
||||||
|
},
|
||||||
|
error: (responseData) => {
|
||||||
|
const message = `Creating request has been failed. Code: ${responseData.status}. Message: ${responseData.responseText || responseData.statusText}`;
|
||||||
|
error(message);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
delete(modelId, success, error) {
|
||||||
|
$.ajax({
|
||||||
|
url: `/auto_annotation/delete/${modelId}`,
|
||||||
|
type: 'DELETE',
|
||||||
|
success,
|
||||||
|
error: (data) => {
|
||||||
|
const message = `Deleting request has been failed. Code: ${data.status}. Message: ${data.responseText || data.statusText}`;
|
||||||
|
error(message);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
check(workerId, success, error, progress) {
|
||||||
|
function updateProgress(data) {
|
||||||
|
if (data.progress && progress) {
|
||||||
|
progress(data.progress);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function checkCallback() {
|
||||||
|
$.ajax({
|
||||||
|
url: `/auto_annotation/check/${workerId}`,
|
||||||
|
type: 'GET',
|
||||||
|
success: (data) => {
|
||||||
|
updateProgress(data, progress);
|
||||||
|
|
||||||
|
switch (data.status) {
|
||||||
|
case 'failed':
|
||||||
|
error(`Checking request has returned the "${data.status}" status. Message: ${data.error}`);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'unknown':
|
||||||
|
error(`Checking request has returned the "${data.status}" status.`);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'finished':
|
||||||
|
success();
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
setTimeout(checkCallback, 1000);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
error: (data) => {
|
||||||
|
const message = `Checking request has been failed. Code: ${data.status}. Message: ${data.responseText || data.statusText}`;
|
||||||
|
error(message);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
setTimeout(checkCallback, 1000);
|
||||||
|
},
|
||||||
|
|
||||||
|
meta(tids, success, error) {
|
||||||
|
$.ajax({
|
||||||
|
url: '/auto_annotation/meta/get',
|
||||||
|
type: 'POST',
|
||||||
|
data: JSON.stringify(tids),
|
||||||
|
contentType: 'application/json',
|
||||||
|
success,
|
||||||
|
error: (data) => {
|
||||||
|
const message = `Getting meta request has been failed. Code: ${data.status}. Message: ${data.responseText || data.statusText}`;
|
||||||
|
error(message);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
cancel(tid, success, error) {
|
||||||
|
$.ajax({
|
||||||
|
url: `/auto_annotation/cancel/${tid}`,
|
||||||
|
type: 'GET',
|
||||||
|
success,
|
||||||
|
error: (data) => {
|
||||||
|
const message = `Getting meta request has been failed. Code: ${data.status}. Message: ${data.responseText || data.statusText}`;
|
||||||
|
error(message);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
class AutoAnnotationModelManagerView {
|
||||||
|
constructor() {
|
||||||
|
const html = `<div class="modal hidden" id="${window.cvat.autoAnnotation.managerWindowId}">
|
||||||
|
<div class="modal-content" id="${window.cvat.autoAnnotation.managerContentId}">
|
||||||
|
<div style="float: left; width: 55%; height: 100%;">
|
||||||
|
<center>
|
||||||
|
<label class="regular h1"> Uploaded Models </label>
|
||||||
|
</center>
|
||||||
|
<div style="overflow: auto; height: 90%; margin-top: 2%;">
|
||||||
|
<table class="regular modelsTable">
|
||||||
|
<thead>
|
||||||
|
<tr>
|
||||||
|
<th> Name </th>
|
||||||
|
<th> Upload Date </th>
|
||||||
|
<th> Actions </th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody id="${window.cvat.autoAnnotation.managerUploadedModelsId}"> </tbody>
|
||||||
|
</table>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="regular" id="${window.cvat.autoAnnotation.uploadContentId}">
|
||||||
|
<center>
|
||||||
|
<label class="regular h1" id="${window.cvat.autoAnnotation.uploadTitleId}"> Create Model </label>
|
||||||
|
</center>
|
||||||
|
<table>
|
||||||
|
<tr>
|
||||||
|
<td style="width: 25%"> <label class="regular h3"> Name: </label> </td>
|
||||||
|
<td> <input type="text" id="${window.cvat.autoAnnotation.uploadNameInputId}" class="regular h3" style="width: 100%"> </td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td> <label class="regular h3"> Source: </label> </td>
|
||||||
|
<td>
|
||||||
|
<input id="${window.cvat.autoAnnotation.uploadLocalSourceId}" type="radio" name="modelSourceType" value="local" checked>
|
||||||
|
<label for="${window.cvat.autoAnnotation.uploadLocalSourceId}" class="regular h3"> Local </label>
|
||||||
|
<br>
|
||||||
|
<input id="${window.cvat.autoAnnotation.uploadShareSourceId}" type="radio" name="modelSourceType" value="shared">
|
||||||
|
<label for="${window.cvat.autoAnnotation.uploadShareSourceId}" class="regular h3"> Share </label>
|
||||||
|
</td>
|
||||||
|
</tr>
|
||||||
|
<tr id="${window.cvat.autoAnnotation.uploadGloballyBlockId}">
|
||||||
|
<td> <label class="regular h3"> Upload Globally </label> </td>
|
||||||
|
<td> <input type="checkbox" id="${window.cvat.autoAnnotation.uploadGloballyId}"> </td>
|
||||||
|
</tr>
|
||||||
|
</table>
|
||||||
|
<div style="text-align: left;">
|
||||||
|
<div>
|
||||||
|
<button id="${window.cvat.autoAnnotation.selectFilesButtonId}" class="regular h3"> Select Files </button>
|
||||||
|
<label id="${window.cvat.autoAnnotation.selectedFilesId}" class="regular h3" style="margin-left: 10px"> No Files </label>
|
||||||
|
<input id="${window.cvat.autoAnnotation.localFileSelectorId}" type="file" accept=".bin,.xml,.json,.py" style="display: none" multiple>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<div style="float: right; width: 50%; height: 50px;">
|
||||||
|
<button class="regular h3" id="${window.cvat.autoAnnotation.submitUploadButtonId}"> Submit </button>
|
||||||
|
<button class="regular h3" id="${window.cvat.autoAnnotation.cancelUploadButtonId}"> Cancel </button>
|
||||||
|
</div>
|
||||||
|
<div style="float: left; overflow-y: auto; height: 75px; overflow: auto; width: 100%; word-break: break-word;">
|
||||||
|
<label class="regular h3 selectable" style="float: left;" id="${window.cvat.autoAnnotation.uploadMessageId}"> </label>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>`;
|
||||||
|
|
||||||
|
this.el = $(html);
|
||||||
|
|
||||||
|
this.table = this.el.find(`#${window.cvat.autoAnnotation.managerUploadedModelsId}`);
|
||||||
|
this.globallyBlock = this.el.find(`#${window.cvat.autoAnnotation.uploadGloballyBlockId}`);
|
||||||
|
this.uploadTitle = this.el.find(`#${window.cvat.autoAnnotation.uploadTitleId}`);
|
||||||
|
this.uploadNameInput = this.el.find(`#${window.cvat.autoAnnotation.uploadNameInputId}`);
|
||||||
|
this.uploadMessage = this.el.find(`#${window.cvat.autoAnnotation.uploadMessageId}`);
|
||||||
|
this.selectedFilesLabel = this.el.find(`#${window.cvat.autoAnnotation.selectedFilesId}`);
|
||||||
|
this.modelNameInput = this.el.find(`#${window.cvat.autoAnnotation.uploadNameInputId}`);
|
||||||
|
this.localSource = this.el.find(`#${window.cvat.autoAnnotation.uploadLocalSourceId}`);
|
||||||
|
this.shareSource = this.el.find(`#${window.cvat.autoAnnotation.uploadShareSourceId}`);
|
||||||
|
this.cancelButton = this.el.find(`#${window.cvat.autoAnnotation.cancelUploadButtonId}`);
|
||||||
|
this.submitButton = this.el.find(`#${window.cvat.autoAnnotation.submitUploadButtonId}`);
|
||||||
|
this.globallyBox = this.el.find(`#${window.cvat.autoAnnotation.uploadGloballyId}`);
|
||||||
|
this.selectButton = this.el.find(`#${window.cvat.autoAnnotation.selectFilesButtonId}`);
|
||||||
|
this.localSelector = this.el.find(`#${window.cvat.autoAnnotation.localFileSelectorId}`);
|
||||||
|
this.shareSelector = $('#dashboardShareBrowseModal');
|
||||||
|
this.shareBrowseTree = $('#dashboardShareBrowser');
|
||||||
|
this.submitShare = $('#dashboardSubmitBrowseServer');
|
||||||
|
|
||||||
|
this.id = null;
|
||||||
|
this.source = this.localSource.prop('checked') ? 'local' : 'shared';
|
||||||
|
this.files = [];
|
||||||
|
|
||||||
|
function filesLabel(source, files) {
|
||||||
|
const fileLabels = source === 'local' ? [...files].map(el => el.name) : files;
|
||||||
|
if (fileLabels.length) {
|
||||||
|
const labelStr = fileLabels.join(', ');
|
||||||
|
if (labelStr.length > 30) {
|
||||||
|
return `${labelStr.substr(0, 30)}..`;
|
||||||
|
}
|
||||||
|
|
||||||
|
return labelStr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 'No Files';
|
||||||
|
}
|
||||||
|
|
||||||
|
function extractFiles(extensions, files, source) {
|
||||||
|
const extractedFiles = {};
|
||||||
|
function getExt(file) {
|
||||||
|
return source === 'local' ? file.name.split('.').pop() : file.split('.').pop();
|
||||||
|
}
|
||||||
|
|
||||||
|
function addFile(file, extention) {
|
||||||
|
if (extention in files) {
|
||||||
|
throw Error(`More than one file with the extension .${extention} have been found`);
|
||||||
|
}
|
||||||
|
|
||||||
|
extractedFiles[extention] = file;
|
||||||
|
}
|
||||||
|
|
||||||
|
files.forEach((file) => {
|
||||||
|
const fileExt = getExt(file);
|
||||||
|
if (extensions.includes(fileExt)) {
|
||||||
|
addFile(file, fileExt);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return extractedFiles;
|
||||||
|
}
|
||||||
|
|
||||||
|
function validateFiles(isUpdate, files, source) {
|
||||||
|
const extensions = ['xml', 'bin', 'py', 'json'];
|
||||||
|
const extractedFiles = extractFiles(extensions, files, source);
|
||||||
|
|
||||||
|
if (!isUpdate) {
|
||||||
|
extensions.forEach((extension) => {
|
||||||
|
if (!(extension in extractedFiles)) {
|
||||||
|
throw Error(`Please specify a .${extension} file`);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return extractedFiles;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.localSource.on('click', () => {
|
||||||
|
if (this.source !== 'local') {
|
||||||
|
this.source = 'local';
|
||||||
|
this.files = [];
|
||||||
|
this.selectedFilesLabel.text(filesLabel(this.source, this.files));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
this.shareSource.on('click', () => {
|
||||||
|
if (this.source !== 'shared') {
|
||||||
|
this.source = 'shared';
|
||||||
|
this.files = [];
|
||||||
|
this.selectedFilesLabel.text(filesLabel(this.source, this.files));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
this.selectButton.on('click', () => {
|
||||||
|
if (this.source === 'local') {
|
||||||
|
this.localSelector.click();
|
||||||
|
} else {
|
||||||
|
this.shareSelector.appendTo('body');
|
||||||
|
this.shareBrowseTree.jstree('refresh');
|
||||||
|
this.shareSelector.removeClass('hidden');
|
||||||
|
this.shareBrowseTree.jstree({
|
||||||
|
core: {
|
||||||
|
data: {
|
||||||
|
url: 'get_share_nodes',
|
||||||
|
data: node => ({ id: node.id }),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
plugins: ['checkbox', 'sort'],
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
this.submitShare.on('click', () => {
|
||||||
|
if (!this.el.hasClass('hidden')) {
|
||||||
|
this.shareSelector.addClass('hidden');
|
||||||
|
this.files = this.shareBrowseTree.jstree(true).get_selected();
|
||||||
|
this.selectedFilesLabel.text(filesLabel(this.source, this.files));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
this.localSelector.on('change', (e) => {
|
||||||
|
this.files = Array.from(e.target.files);
|
||||||
|
this.selectedFilesLabel.text(filesLabel(this.source, this.files));
|
||||||
|
});
|
||||||
|
|
||||||
|
this.cancelButton.on('click', () => this.el.addClass('hidden'));
|
||||||
|
this.submitButton.on('click', () => {
|
||||||
|
try {
|
||||||
|
this.submitButton.prop('disabled', true);
|
||||||
|
|
||||||
|
const name = $.trim(this.modelNameInput.prop('value'));
|
||||||
|
if (!name.length) {
|
||||||
|
this.uploadMessage.css('color', 'red');
|
||||||
|
this.uploadMessage.text('Please specify a model name');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let validatedFiles = {};
|
||||||
|
try {
|
||||||
|
validatedFiles = validateFiles(this.id !== null, this.files, this.source);
|
||||||
|
} catch (err) {
|
||||||
|
this.uploadMessage.css('color', 'red');
|
||||||
|
this.uploadMessage.text(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const modelData = new FormData();
|
||||||
|
modelData.append('name', name);
|
||||||
|
modelData.append('storage', this.source);
|
||||||
|
modelData.append('shared', this.globallyBox.prop('checked'));
|
||||||
|
|
||||||
|
['xml', 'bin', 'json', 'py'].filter(e => e in validatedFiles).forEach((ext) => {
|
||||||
|
modelData.append(ext, validatedFiles[ext]);
|
||||||
|
});
|
||||||
|
|
||||||
|
this.uploadMessage.text('');
|
||||||
|
const overlay = showOverlay('Send request to the server..');
|
||||||
|
window.cvat.autoAnnotation.server.update(modelData, () => {
|
||||||
|
window.location.reload();
|
||||||
|
}, (message) => {
|
||||||
|
overlay.remove();
|
||||||
|
showMessage(message);
|
||||||
|
}, (progress) => {
|
||||||
|
overlay.setMessage(progress);
|
||||||
|
}, window.cvat.autoAnnotation.server.check, this.id);
|
||||||
|
} finally {
|
||||||
|
this.submitButton.prop('disabled', false);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
reset() {
|
||||||
|
const setBlocked = () => {
|
||||||
|
if (window.cvat.autoAnnotation.data.admin) {
|
||||||
|
this.globallyBlock.removeClass('hidden');
|
||||||
|
} else {
|
||||||
|
this.globallyBlock.addClass('hidden');
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
setBlocked();
|
||||||
|
this.uploadTitle.text('Create Model');
|
||||||
|
this.uploadNameInput.prop('value', '');
|
||||||
|
this.uploadMessage.css('color', '');
|
||||||
|
this.uploadMessage.text('');
|
||||||
|
this.selectedFilesLabel.text('No Files');
|
||||||
|
this.localSource.prop('checked', true);
|
||||||
|
this.globallyBox.prop('checked', false);
|
||||||
|
this.table.empty();
|
||||||
|
|
||||||
|
this.id = null;
|
||||||
|
this.source = this.localSource.prop('checked') ? 'local' : 'share';
|
||||||
|
this.files = [];
|
||||||
|
|
||||||
|
const updateButtonClickHandler = (event) => {
|
||||||
|
this.reset();
|
||||||
|
|
||||||
|
this.uploadTitle.text('Update Model');
|
||||||
|
this.uploadNameInput.prop('value', `${event.data.model.name}`);
|
||||||
|
this.id = event.data.model.id;
|
||||||
|
};
|
||||||
|
|
||||||
|
const deleteButtonClickHandler = (event) => {
|
||||||
|
userConfirm(`Do you actually want to delete the "${event.data.model.name}" model. Are you sure?`, () => {
|
||||||
|
window.cvat.autoAnnotation.server.delete(event.data.model.id, () => {
|
||||||
|
const filtered = window.cvat.autoAnnotation.data.models.filter(
|
||||||
|
item => item !== event.data.model,
|
||||||
|
);
|
||||||
|
window.cvat.autoAnnotation.data.models = filtered;
|
||||||
|
this.reset();
|
||||||
|
}, (message) => {
|
||||||
|
showMessage(message);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const getModelModifyButtons = (model) => {
|
||||||
|
if (model.primary) {
|
||||||
|
return '<td> <label class="h1 regular"> Primary Model </label> </td>';
|
||||||
|
}
|
||||||
|
|
||||||
|
const updateButtonHtml = '<button class="regular h3" style="width: 7em;"> Update </button>';
|
||||||
|
const deleteButtonHtml = '<button class="regular h3" style="width: 7em; margin-top: 5%;"> Delete </button>';
|
||||||
|
|
||||||
|
return $('<td> </td>').append(
|
||||||
|
$(updateButtonHtml).on('click', { model }, updateButtonClickHandler),
|
||||||
|
$(deleteButtonHtml).on('click', { model }, deleteButtonClickHandler),
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
window.cvat.autoAnnotation.data.models.forEach((model) => {
|
||||||
|
const rowHtml = `<tr>
|
||||||
|
<td> ${model.name} </td>
|
||||||
|
<td> ${model.uploadDate} </td>
|
||||||
|
</tr>`;
|
||||||
|
|
||||||
|
this.table.append(
|
||||||
|
$(rowHtml).append(getModelModifyButtons(model)),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
show() {
|
||||||
|
this.el.removeClass('hidden');
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
get element() {
|
||||||
|
return this.el;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class AutoAnnotationModelRunnerView {
|
||||||
|
constructor() {
|
||||||
|
const html = `<div class="modal hidden" id="${window.cvat.autoAnnotation.runnerWindowId}">
|
||||||
|
<div class="modal-content" id="${window.cvat.autoAnnotation.runnerContentId}">
|
||||||
|
<div style="width: 55%; height: 100%; float: left;">
|
||||||
|
<center style="height: 10%;">
|
||||||
|
<label class="regular h1"> Uploaded Models </label>
|
||||||
|
</center>
|
||||||
|
<div style="height: 70%; overflow: auto; margin-top: 2%;">
|
||||||
|
<table class="modelsTable" id="${window.cvat.autoAnnotation.runnerUploadedModelsId}"> </table>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<input type="checkbox" id="${window.cvat.autoAnnotation.removeCurrentAnnotationId}"/>
|
||||||
|
<label class="regular h3" for="${window.cvat.autoAnnotation.removeCurrentAnnotationId}"> Remove current annotation </label>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div style="width: 40%; height: 100%; float: left; margin-left: 3%;">
|
||||||
|
<center style="height: 10%;">
|
||||||
|
<label class="regular h1"> Annotation Labels </label>
|
||||||
|
</center>
|
||||||
|
<div style="height: 70%; overflow: auto; margin-top: 2%;">
|
||||||
|
<table class="regular" style="text-align: center; word-break: break-all; width: 100%;">
|
||||||
|
<thead>
|
||||||
|
<tr style="width: 100%;">
|
||||||
|
<th style="width: 45%;"> Task Label </th>
|
||||||
|
<th style="width: 45%;"> DL Model Label </th>
|
||||||
|
<th style="width: 10%;"> </th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody id="${window.cvat.autoAnnotation.annotationLabelsId}">
|
||||||
|
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
</div>
|
||||||
|
<div style="float:right;">
|
||||||
|
<button class="regular h3" style="width: 6em;" id="${window.cvat.autoAnnotation.submitAnnotationId}"> Start </button>
|
||||||
|
<button class="regular h3" style="width: 6em;" id="${window.cvat.autoAnnotation.cancelAnnotationId}"> Cancel </button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>`;
|
||||||
|
|
||||||
|
this.el = $(html);
|
||||||
|
this.id = null;
|
||||||
|
this.tid = null;
|
||||||
|
this.initButton = null;
|
||||||
|
this.modelsTable = this.el.find(`#${window.cvat.autoAnnotation.runnerUploadedModelsId}`);
|
||||||
|
this.labelsTable = this.el.find(`#${window.cvat.autoAnnotation.annotationLabelsId}`);
|
||||||
|
this.active = null;
|
||||||
|
|
||||||
|
this.el.find(`#${window.cvat.autoAnnotation.cancelAnnotationId}`).on('click', () => {
|
||||||
|
this.el.addClass('hidden');
|
||||||
|
});
|
||||||
|
|
||||||
|
this.el.find(`#${window.cvat.autoAnnotation.submitAnnotationId}`).on('click', () => {
|
||||||
|
try {
|
||||||
|
if (this.id === null) {
|
||||||
|
throw Error('Please specify a model for an annotation process');
|
||||||
|
}
|
||||||
|
|
||||||
|
const mapping = {};
|
||||||
|
$('.annotatorMappingRow').each((_, element) => {
|
||||||
|
const dlModelLabel = $(element).find('.annotatorDlLabelSelector')[0].value;
|
||||||
|
const taskLabel = $(element).find('.annotatorTaskLabelSelector')[0].value;
|
||||||
|
if (dlModelLabel in mapping) {
|
||||||
|
throw Error(`The label "${dlModelLabel}" has been specified twice or more`);
|
||||||
|
}
|
||||||
|
mapping[dlModelLabel] = taskLabel;
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!Object.keys(mapping).length) {
|
||||||
|
throw Error('Labels for an annotation process haven\'t been found');
|
||||||
|
}
|
||||||
|
|
||||||
|
const overlay = showOverlay('Request has been sent');
|
||||||
|
window.cvat.autoAnnotation.server.start(this.id, this.tid, {
|
||||||
|
reset: $(`#${window.cvat.autoAnnotation.removeCurrentAnnotationId}`).prop('checked'),
|
||||||
|
labels: mapping,
|
||||||
|
}, () => {
|
||||||
|
overlay.remove();
|
||||||
|
this.initButton[0].setupRun();
|
||||||
|
window.cvat.autoAnnotation.runner.hide();
|
||||||
|
}, (message) => {
|
||||||
|
overlay.remove();
|
||||||
|
this.initButton[0].setupRun();
|
||||||
|
showMessage(message);
|
||||||
|
}, () => {
|
||||||
|
window.location.reload();
|
||||||
|
}, window.cvat.autoAnnotation.server.check);
|
||||||
|
} catch (error) {
|
||||||
|
showMessage(error);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
reset(data, initButton) {
|
||||||
|
function labelsSelect(labels, elClass) {
|
||||||
|
const select = $(`<select class="regular h3 ${elClass}" style="width:100%;"> </select>`);
|
||||||
|
labels.forEach(label => select.append($(`<option value="${label}"> ${label} </option>`)));
|
||||||
|
select.prop('value', null);
|
||||||
|
|
||||||
|
return select;
|
||||||
|
}
|
||||||
|
|
||||||
|
function makeCreator(dlSelect, taskSelect, callback) {
|
||||||
|
let dlIsFilled = false;
|
||||||
|
let taskIsFilled = false;
|
||||||
|
const creator = $('<tr style="margin-bottom: 5px;"> </tr>').append(
|
||||||
|
$('<td style="width: 45%;"> </td>').append(taskSelect),
|
||||||
|
$('<td style="width: 45%;"> </td>').append(dlSelect),
|
||||||
|
);
|
||||||
|
|
||||||
|
const onSelectHandler = () => {
|
||||||
|
$('<td style="width: 10%; position: relative;"> </td>').append(
|
||||||
|
$('<a class="close"></a>').css('top', '0px').on('click', (e) => {
|
||||||
|
$(e.target.parentNode.parentNode).remove();
|
||||||
|
}),
|
||||||
|
).appendTo(creator);
|
||||||
|
|
||||||
|
creator.addClass('annotatorMappingRow');
|
||||||
|
callback();
|
||||||
|
};
|
||||||
|
|
||||||
|
dlSelect.on('change', (e) => {
|
||||||
|
if (e.target.value && taskIsFilled) {
|
||||||
|
dlSelect.off('change');
|
||||||
|
taskSelect.off('change');
|
||||||
|
onSelectHandler();
|
||||||
|
}
|
||||||
|
dlIsFilled = Boolean(e.target.value);
|
||||||
|
});
|
||||||
|
|
||||||
|
taskSelect.on('change', (e) => {
|
||||||
|
if (e.target.value && dlIsFilled) {
|
||||||
|
dlSelect.off('change');
|
||||||
|
taskSelect.off('change');
|
||||||
|
onSelectHandler();
|
||||||
|
}
|
||||||
|
|
||||||
|
taskIsFilled = Boolean(e.target.value);
|
||||||
|
});
|
||||||
|
|
||||||
|
return creator;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.id = null;
|
||||||
|
this.initButton = initButton;
|
||||||
|
this.tid = data.taskid;
|
||||||
|
this.modelsTable.empty();
|
||||||
|
this.labelsTable.empty();
|
||||||
|
this.active = null;
|
||||||
|
|
||||||
|
const modelItemClickHandler = (event) => {
|
||||||
|
if (this.active) {
|
||||||
|
this.active.style.color = '';
|
||||||
|
}
|
||||||
|
|
||||||
|
this.id = event.data.model.id;
|
||||||
|
this.active = event.target;
|
||||||
|
this.active.style.color = 'darkblue';
|
||||||
|
|
||||||
|
this.labelsTable.empty();
|
||||||
|
const labels = Object.values(event.data.data.spec.labels);
|
||||||
|
const intersection = labels.filter(el => event.data.model.labels.indexOf(el) !== -1);
|
||||||
|
intersection.forEach((label) => {
|
||||||
|
const dlSelect = labelsSelect(event.data.model.labels, 'annotatorDlLabelSelector');
|
||||||
|
dlSelect.prop('value', label);
|
||||||
|
const taskSelect = labelsSelect(labels, 'annotatorTaskLabelSelector');
|
||||||
|
taskSelect.prop('value', label);
|
||||||
|
$('<tr class="annotatorMappingRow" style="margin-bottom: 5px;"> </tr>').append(
|
||||||
|
$('<td style="width: 45%;"> </td>').append(taskSelect),
|
||||||
|
$('<td style="width: 45%;"> </td>').append(dlSelect),
|
||||||
|
$('<td style="width: 10%; position: relative;"> </td>').append(
|
||||||
|
$('<a class="close"></a>').css('top', '0px').on('click', (e) => {
|
||||||
|
$(e.target.parentNode.parentNode).remove();
|
||||||
|
}),
|
||||||
|
),
|
||||||
|
).appendTo(this.labelsTable);
|
||||||
|
});
|
||||||
|
|
||||||
|
const dlSelect = labelsSelect(event.data.model.labels, 'annotatorDlLabelSelector');
|
||||||
|
const taskSelect = labelsSelect(labels, 'annotatorTaskLabelSelector');
|
||||||
|
|
||||||
|
const callback = () => {
|
||||||
|
makeCreator(
|
||||||
|
labelsSelect(event.data.model.labels, 'annotatorDlLabelSelector'),
|
||||||
|
labelsSelect(labels, 'annotatorTaskLabelSelector'),
|
||||||
|
callback,
|
||||||
|
).appendTo(this.labelsTable);
|
||||||
|
};
|
||||||
|
|
||||||
|
makeCreator(dlSelect, taskSelect, callback).appendTo(this.labelsTable);
|
||||||
|
};
|
||||||
|
|
||||||
|
window.cvat.autoAnnotation.data.models.forEach((model) => {
|
||||||
|
this.modelsTable.append(
|
||||||
|
$(`<tr> <td> <label class="regular h3"> ${model.name} (${model.uploadDate}) </label> </td> </tr>`).on(
|
||||||
|
'click', { model, data }, modelItemClickHandler,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
show() {
|
||||||
|
this.el.removeClass('hidden');
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
hide() {
|
||||||
|
this.el.addClass('hidden');
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
get element() {
|
||||||
|
return this.el;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
window.cvat.autoAnnotation = {
|
||||||
|
managerWindowId: 'annotatorManagerWindow',
|
||||||
|
managerContentId: 'annotatorManagerContent',
|
||||||
|
managerUploadedModelsId: 'annotatorManagerUploadedModels',
|
||||||
|
uploadContentId: 'annotatorManagerUploadModel',
|
||||||
|
uploadTitleId: 'annotatorManagerUploadTitle',
|
||||||
|
uploadNameInputId: 'annotatorManagerUploadNameInput',
|
||||||
|
uploadLocalSourceId: 'annotatorManagerUploadLocalSource',
|
||||||
|
uploadShareSourceId: 'annotatorManagerUploadShareSource',
|
||||||
|
uploadGloballyId: 'annotatorManagerUploadGlobally',
|
||||||
|
uploadGloballyBlockId: 'annotatorManagerUploadGloballyblock',
|
||||||
|
selectFilesButtonId: 'annotatorManagerUploadSelector',
|
||||||
|
selectedFilesId: 'annotatorManagerUploadSelectedFiles',
|
||||||
|
localFileSelectorId: 'annotatorManagerUploadLocalSelector',
|
||||||
|
shareFileSelectorId: 'annotatorManagerUploadShareSelector',
|
||||||
|
submitUploadButtonId: 'annotatorManagerSubmitUploadButton',
|
||||||
|
cancelUploadButtonId: 'annotatorManagerCancelUploadButton',
|
||||||
|
uploadMessageId: 'annotatorUploadStatusMessage',
|
||||||
|
|
||||||
|
runnerWindowId: 'annotatorRunnerWindow',
|
||||||
|
runnerContentId: 'annotatorRunnerContent',
|
||||||
|
runnerUploadedModelsId: 'annotatorRunnerUploadedModels',
|
||||||
|
removeCurrentAnnotationId: 'annotatorRunnerRemoveCurrentAnnotationBox',
|
||||||
|
annotationLabelsId: 'annotatorRunnerAnnotationLabels',
|
||||||
|
submitAnnotationId: 'annotatorRunnerSubmitAnnotationButton',
|
||||||
|
cancelAnnotationId: 'annotatorRunnerCancelAnnotationButton',
|
||||||
|
|
||||||
|
managerButtonId: 'annotatorManagerButton',
|
||||||
|
};
|
||||||
|
|
||||||
|
window.cvat.dashboard.uiCallbacks.push((newElements) => {
|
||||||
|
window.cvat.autoAnnotation.server = AutoAnnotationServer;
|
||||||
|
window.cvat.autoAnnotation.manager = new AutoAnnotationModelManagerView();
|
||||||
|
window.cvat.autoAnnotation.runner = new AutoAnnotationModelRunnerView();
|
||||||
|
|
||||||
|
const tids = Array.from(newElements, el => el.id.split('_')[1]);
|
||||||
|
|
||||||
|
window.cvat.autoAnnotation.server.meta(tids, (data) => {
|
||||||
|
window.cvat.autoAnnotation.data = data;
|
||||||
|
$('body').append(window.cvat.autoAnnotation.manager.element, window.cvat.autoAnnotation.runner.element);
|
||||||
|
$(`<button id="${window.cvat.autoAnnotation.managerButtonId}" class="regular h1" style=""> Model Manager</button>`)
|
||||||
|
.on('click', () => {
|
||||||
|
const overlay = showOverlay('The manager are being setup..');
|
||||||
|
window.cvat.autoAnnotation.manager.reset().show();
|
||||||
|
overlay.remove();
|
||||||
|
}).appendTo('#dashboardManageButtons');
|
||||||
|
|
||||||
|
newElements.each((_, element) => {
|
||||||
|
const elem = $(element);
|
||||||
|
const tid = +elem.attr('id').split('_')[1];
|
||||||
|
|
||||||
|
const button = $('<button> Run Auto Annotation </button>').addClass('regular dashboardButtonUI');
|
||||||
|
button[0].setupRun = function setupRun() {
|
||||||
|
const self = $(this);
|
||||||
|
self.text('Run Auto Annotation').off('click').on('click', () => {
|
||||||
|
const overlay = showOverlay('Task date are being recieved from the server..');
|
||||||
|
$.ajax({
|
||||||
|
url: `/get/task/${tid}`,
|
||||||
|
dataType: 'json',
|
||||||
|
success: (responseData) => {
|
||||||
|
overlay.setMessage('The model runner are being setup..');
|
||||||
|
window.cvat.autoAnnotation.runner.reset(responseData, self).show();
|
||||||
|
overlay.remove();
|
||||||
|
},
|
||||||
|
error: (responseData) => {
|
||||||
|
showMessage(`Can't get task data. Code: ${responseData.status}. Message: ${responseData.responseText || responseData.statusText}`);
|
||||||
|
},
|
||||||
|
complete: () => {
|
||||||
|
overlay.remove();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
button[0].setupCancel = function setupCancel() {
|
||||||
|
const self = $(this);
|
||||||
|
self.off('click').text('Cancel Auto Annotation').on('click', () => {
|
||||||
|
userConfirm('Process will be canceled. Are you sure?', () => {
|
||||||
|
window.cvat.autoAnnotation.server.cancel(tid, () => {
|
||||||
|
this.setupRun();
|
||||||
|
}, (message) => {
|
||||||
|
showMessage(message);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
window.cvat.autoAnnotation.server.check(
|
||||||
|
window.cvat.autoAnnotation.data.run[tid].rq_id,
|
||||||
|
() => {
|
||||||
|
this.setupRun();
|
||||||
|
},
|
||||||
|
(error) => {
|
||||||
|
button[0].setupRun();
|
||||||
|
button.text('Annotation has failed');
|
||||||
|
button.title(error);
|
||||||
|
},
|
||||||
|
(progress) => {
|
||||||
|
button.text(`Cancel Auto Annotation (${progress.toString().slice(0, 4)})%`);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
const taskStatus = window.cvat.autoAnnotation.data.run[tid];
|
||||||
|
if (taskStatus && ['queued', 'started'].includes(taskStatus.status)) {
|
||||||
|
button[0].setupCancel();
|
||||||
|
} else {
|
||||||
|
button[0].setupRun();
|
||||||
|
}
|
||||||
|
|
||||||
|
button.appendTo(elem.find('div.dashboardButtonsUI')[0]);
|
||||||
|
});
|
||||||
|
}, (error) => {
|
||||||
|
showMessage(`Cannot get models meta information: ${error}`);
|
||||||
|
});
|
||||||
|
});
|
||||||
@ -0,0 +1,83 @@
|
|||||||
|
#annotatorManagerContent, #annotatorRunnerContent {
|
||||||
|
width: 800px;
|
||||||
|
height: 300px;
|
||||||
|
}
|
||||||
|
|
||||||
|
#annotatorManagerButton {
|
||||||
|
padding: 7px;
|
||||||
|
margin-left: 4px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable {
|
||||||
|
width: 100%;
|
||||||
|
color:#666;
|
||||||
|
text-shadow: 1px 1px 0px #fff;
|
||||||
|
background:#D2D3D4;
|
||||||
|
border:#ccc 1px solid;
|
||||||
|
border-radius: 3px;
|
||||||
|
box-shadow: 0 1px 2px black;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable th {
|
||||||
|
border-top: 1px solid #fafafa;
|
||||||
|
border-bottom: 1px solid #e0e0e0;
|
||||||
|
background: #ededed;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable th:first-child {
|
||||||
|
text-align: left;
|
||||||
|
padding-left:20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr:first-child th:first-child {
|
||||||
|
border-top-left-radius:3px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr:first-child th:last-child {
|
||||||
|
border-top-right-radius:3px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr {
|
||||||
|
text-align: center;
|
||||||
|
padding-left: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable td:first-child {
|
||||||
|
text-align: left;
|
||||||
|
padding-left: 20px;
|
||||||
|
border-left: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable td {
|
||||||
|
padding: 18px;
|
||||||
|
border-top: 1px solid #ffffff;
|
||||||
|
border-bottom:1px solid #e0e0e0;
|
||||||
|
border-left: 1px solid #e0e0e0;
|
||||||
|
background: #fafafa;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr.even td {
|
||||||
|
background: #f6f6f6;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr:last-child td {
|
||||||
|
border-bottom:0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr:last-child td:first-child {
|
||||||
|
border-bottom-left-radius:3px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr:last-child td:last-child {
|
||||||
|
border-bottom-right-radius:3px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.modelsTable tr:hover td {
|
||||||
|
background: #f2f2f2;
|
||||||
|
}
|
||||||
|
|
||||||
|
#annotatorManagerUploadModel {
|
||||||
|
float: left;
|
||||||
|
padding-left: 3%;
|
||||||
|
width: 40%;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue