def create_annotation(form, object_name, labels, frame):
annotation = Region()
annotation.object_name = object_name
if form.cleaned_data['high_level']:
annotation.full_frame = True
annotation.x = 0
annotation.y = 0
annotation.h = 0
annotation.w = 0
else:
annotation.full_frame = False
annotation.x = form.cleaned_data['x']
annotation.y = form.cleaned_data['y']
annotation.h = form.cleaned_data['h']
annotation.w = form.cleaned_data['w']
annotation.text = form.cleaned_data['text']
annotation.metadata = form.cleaned_data['metadata']
annotation.frame = frame
annotation.video = frame.video
annotation.region_type = Region.ANNOTATION
annotation.save()
for lname in labels:
if lname.strip():
dl, _ = Label.objects.get_or_create(name=lname, set="UI")
rl = RegionLabel()
rl.video = annotation.video
rl.frame = annotation.frame
rl.region = annotation
rl.label = dl
rl.save()
def assign_tags(video_id):
import django
from PIL import Image
sys.path.append(os.path.dirname(__file__))
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "dva.settings")
django.setup()
from django.conf import settings
from dvaapp.models import Video, Frame, Region
from dvalib import entity, annotator
dv = Video.objects.get(id=video_id)
frames = Frame.objects.all().filter(video=dv)
v = entity.WVideo(dvideo=dv, media_dir=settings.MEDIA_ROOT)
wframes = {
df.pk: entity.WFrame(video=v,
frame_index=df.frame_index,
primary_key=df.pk)
for df in frames
}
algorithm = annotator.OpenImagesAnnotator()
logging.info("starting annotation {}".format(algorithm.name))
for k, f in wframes.items():
tags = algorithm.apply(f.local_path())
a = Region()
a.region_type = Region.ANNOTATION
a.frame_id = k
a.video_id = video_id
a.object_name = "OpenImagesTag"
a.metadata_text = " ".join([t for t, v in tags.iteritems() if v > 0.1])
a.metadata_json = json.dumps(
{t: 100.0 * v
for t, v in tags.iteritems() if v > 0.1})
a.full_frame = True
a.save()
print a.metadata_text
def annotate_entire_frame(request, frame_pk):
frame = Frame.objects.get(pk=frame_pk)
annotation = None
if request.method == 'POST':
if request.POST.get('text').strip() \
or request.POST.get('metadata').strip() \
or request.POST.get('object_name', None):
annotation = Region()
annotation.region_type = Region.ANNOTATION
annotation.x = 0
annotation.y = 0
annotation.h = 0
annotation.w = 0
annotation.full_frame = True
annotation.text = request.POST.get('text')
annotation.metadata = request.POST.get('metadata')
annotation.object_name = request.POST.get('object_name',
'frame_metadata')
annotation.frame = frame
annotation.video = frame.video
annotation.save()
for label_name in request.POST.get('tags').split(','):
if label_name.strip():
if annotation:
dl = RegionLabel()
dl.video = frame.video
dl.frame = frame
dl.label = Label.objects.get_or_create(name=label_name,
set="UI")[0]
dl.region = annotation
dl.save()
else:
dl = FrameLabel()
dl.video = frame.video
dl.frame = frame
dl.label = Label.objects.get_or_create(name=label_name,
set="UI")[0]
dl.save()
return redirect("frame_detail", pk=frame.pk)