Fixed tensorflow annotation without OpenVINO (#359)

main
Boris Sekachev 7 years ago committed by Nikita Manovich
parent ab013b0da1
commit 9fec0c223b

@ -8,7 +8,6 @@ from django.core.exceptions import ObjectDoesNotExist
from django.shortcuts import render
from rules.contrib.views import permission_required, objectgetter
from cvat.apps.authentication.decorators import login_required
from cvat.apps.auto_annotation.inference_engine import make_plugin, make_network
from cvat.apps.engine.models import Task as TaskModel
from cvat.apps.engine import annotation, task
@ -33,6 +32,8 @@ def load_image_into_numpy(image):
def run_inference_engine_annotation(image_list, labels_mapping, treshold):
from cvat.apps.auto_annotation.inference_engine import make_plugin, make_network
def _normalize_box(box, w, h, dw, dh):
xmin = min(int(box[0] * dw * w), w)
ymin = min(int(box[1] * dh * h), h)

Loading…
Cancel
Save