Source code for wbia.control.manual_annot_funcs

# -*- coding: utf-8 -*-
"""
Autogen:
    python -c "import utool as ut; ut.write_modscript_alias('Tgen.sh', 'wbia.templates.template_generator')"  # NOQA
    sh Tgen.sh --key annot --invert --Tcfg with_getters=True with_setters=True --modfname manual_annot_funcs --funcname-filter=age_m  # NOQA
    sh Tgen.sh --key annot --invert --Tcfg with_getters=True with_setters=True --modfname manual_annot_funcs --funcname-filter=is_  # NOQA
    sh Tgen.sh --key annot --invert --Tcfg with_getters=True with_setters=True --modfname manual_annot_funcs --funcname-filter=is_ --diff  # NOQA
"""
import logging
import uuid
import numpy as np
from wbia import constants as const
from wbia.control import accessor_decors, controller_inject
import utool as ut
from wbia.other import ibsfuncs
from wbia.control.controller_inject import make_ibs_register_decorator

# from collections import namedtuple
from wbia.web import routes_ajax
import requests

print, rrr, profile = ut.inject2(__name__)
logger = logging.getLogger('wbia')


CLASS_INJECT_KEY, register_ibs_method = make_ibs_register_decorator(__name__)


register_api = controller_inject.get_wbia_flask_api(__name__)


ANNOT_AGE_MONTHS_EST_MAX = 'annot_age_months_est_max'
ANNOT_AGE_MONTHS_EST_MIN = 'annot_age_months_est_min'
ANNOT_NOTE = 'annot_note'
ANNOT_NUM_VERTS = 'annot_num_verts'
ANNOT_PARENT_ROWID = 'annot_parent_rowid'
ANNOT_ROWID = 'annot_rowid'
ANNOT_TAG_TEXT = 'annot_tag_text'
ANNOT_SEMANTIC_UUID = 'annot_semantic_uuid'
ANNOT_THETA = 'annot_theta'
ANNOT_VERTS = 'annot_verts'
ANNOT_UUID = 'annot_uuid'
ANNOT_YAW = 'annot_yaw'
ANNOT_VIEWPOINT = 'annot_viewpoint'
ANNOT_VISUAL_UUID = 'annot_visual_uuid'
CONFIG_ROWID = 'config_rowid'
FEATWEIGHT_ROWID = 'featweight_rowid'
IMAGE_ROWID = 'image_rowid'
NAME_ROWID = 'name_rowid'
SPECIES_ROWID = 'species_rowid'
ANNOT_EXEMPLAR_FLAG = 'annot_exemplar_flag'
ANNOT_QUALITY = 'annot_quality'
ANNOT_ROWIDS = 'annot_rowids'
GAR_ROWID = 'gar_rowid'
PART_ROWID = 'part_rowid'
ANNOT_STAGED_FLAG = 'annot_staged_flag'
ANNOT_STAGED_USER_ID = 'annot_staged_user_identity'


# Define the properties that will make up the semantic / visual uuids
VISUAL_TUP_PROPS = ['image_uuids', 'verts', 'thetas']
# TODO: Remove viewpoint and species?
SEMANTIC_TUP_PROPS = ['viewpoint_code', 'names', 'species_texts']


[docs]@register_ibs_method @register_api('/api/annot/uuid/', methods=['POST']) def compute_annot_visual_semantic_uuids( ibs, gid_list, include_preprocess=False, **kwargs ): preprocess_dict = ibs._add_annots_preprocess(gid_list, **kwargs) theta_list = preprocess_dict['theta_list'] vert_list = preprocess_dict['vert_list'] species_rowid_list = preprocess_dict['species_rowid_list'] viewpoint_ints = preprocess_dict['viewpoint_ints'] nid_list = preprocess_dict['nid_list'] image_uuid_list = ibs.get_image_uuids(gid_list) species_list = ibs.get_species_texts(species_rowid_list) view_codes = ut.dict_take(ibs.const.VIEW.INT_TO_CODE, viewpoint_ints) name_list = ibs.get_name_texts(nid_list) props = { 'image_uuids': image_uuid_list, 'verts': vert_list, 'thetas': theta_list, 'species_texts': species_list, 'viewpoint_code': view_codes, 'names': name_list, } visual_infotup = ut.take(props, VISUAL_TUP_PROPS) annot_visual_uuid_list = [ut.augment_uuid(*tup) for tup in zip(*visual_infotup)] semantic_infotup = ut.take(props, VISUAL_TUP_PROPS + SEMANTIC_TUP_PROPS) annot_semantic_uuid_list = [ut.augment_uuid(*tup) for tup in zip(*semantic_infotup)] value_dict = { 'visual_uuid_list': annot_visual_uuid_list, 'semantic_uuid_list': annot_semantic_uuid_list, } if include_preprocess: for key in preprocess_dict: assert key not in value_dict value_dict[key] = preprocess_dict[key] return value_dict
@register_ibs_method def _add_annots_preprocess( ibs, gid_list, bbox_list=None, theta_list=None, vert_list=None, species_list=None, species_rowid_list=None, viewpoint_list=None, nid_list=None, name_list=None, skip_cleaning=False, **kwargs ): from vtool import geometry # BBOXES / VERTS # For import only, we can specify both by setting import_override to True assert bool(bbox_list is None) != bool( vert_list is None ), 'must specify exactly one of bbox_list or vert_list' if vert_list is None: vert_list = geometry.verts_list_from_bboxes_list(bbox_list) elif bbox_list is None: bbox_list = geometry.bboxes_from_vert_list(vert_list) if theta_list is None: theta_list = [0.0 for _ in range(len(gid_list))] len_gid = len(gid_list) len_bbox = len(bbox_list) len_vert = len(vert_list) len_theta = len(theta_list) try: assert len_gid == len_bbox, 'bbox and gid are not of same size' assert len_gid == len_theta, 'theta and gid are not of same size' assert len_gid == len_vert, 'vert and gid are not of same size' except AssertionError as ex: ut.printex(ex, key_list=['len_vert', 'len_gid', 'len_bbox' 'len_theta']) raise # SPECIES if species_rowid_list is not None: assert species_list is None, 'cannot mix species_rowid and species' else: if species_list is not None: species_rowid_list = ibs.add_species( species_list, skip_cleaning=skip_cleaning ) else: species_rowid_list = [ const.UNKNOWN_SPECIES_ROWID for _ in range(len(gid_list)) ] # VIEWPOINTS if viewpoint_list is None: viewpoint_list = [None] * len(gid_list) viewpoint_ints = _ensure_viewpoint_to_int(viewpoint_list) # NAMES assert name_list is None or nid_list is None, 'cannot specify both names and nids' if name_list is not None: nid_list = ibs.add_names(name_list) else: if nid_list is None: nid_list = [const.UNKNOWN_NAME_ROWID for _ in range(len(gid_list))] preprocess_dict = { 'bbox_list': bbox_list, 'theta_list': theta_list, 'vert_list': vert_list, 'species_rowid_list': species_rowid_list, 'viewpoint_list': viewpoint_list, 'viewpoint_ints': viewpoint_ints, 'nid_list': nid_list, } return preprocess_dict # ========== # ADDERS # ==========
[docs]@register_ibs_method @accessor_decors.adder @accessor_decors.cache_invalidator( const.IMAGE_TABLE, colnames=[ANNOT_ROWIDS], rowidx=None ) @accessor_decors.cache_invalidator(const.NAME_TABLE, colnames=[ANNOT_ROWIDS]) @accessor_decors.cache_invalidator( const.IMAGESET_TABLE, ['percent_names_with_exemplar_str'] ) @register_api('/api/annot/', methods=['POST']) def add_annots( ibs, gid_list, bbox_list=None, theta_list=None, species_list=None, nid_list=None, name_list=None, vert_list=None, annot_uuid_list=None, yaw_list=None, viewpoint_list=None, quality_list=None, multiple_list=None, interest_list=None, canonical_list=None, detect_confidence_list=None, notes_list=None, annot_visual_uuid_list=None, annot_semantic_uuid_list=None, species_rowid_list=None, staged_uuid_list=None, staged_user_id_list=None, quiet_delete_thumbs=False, prevent_visual_duplicates=True, skip_cleaning=False, delete_thumb=True, **kwargs ): r""" Adds an annotation to images # TODO: remove annot_visual_uuid_list and annot_semantic_uuid_list They are always inferred Args: gid_list (list): image rowids to add annotation to bbox_list (list): of [x, y, w, h] bounding boxes for each image (supply verts instead) theta_list (list): orientations of annotations species_list (list): nid_list (list): name_list (list): detect_confidence_list (list): notes_list (list): vert_list (list): alternative to bounding box annot_uuid_list (list): yaw_list (list): annot_visual_uuid_list (list): annot_semantic_uuid_list (list): quiet_delete_thumbs (bool): Returns: list: aid_list CommandLine: python -m wbia.control.manual_annot_funcs --test-add_annots python -m wbia.control.manual_annot_funcs --test-add_annots --verbose --print-caller Ignore: theta_list = None species_list = None nid_list = None name_list = None detect_confidence_list = None notes_list = None vert_list = None annot_uuid_list = None yaw_list = None quiet_delete_thumbs = False prevent_visual_duplicates = False RESTful: Method: POST URL: /api/annot/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.IBEISControl import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> prevalid = ibs.get_valid_aids() >>> num_add = 2 >>> gid_list = ibs.get_valid_gids()[0:num_add] >>> bbox_list = [(int(w * .1), int(h * .6), int(w * .5), int(h * .3)) ... for (w, h) in ibs.get_image_sizes(gid_list)] >>> # Add a test annotation >>> print('Testing add_annots') >>> aid_list = ibs.add_annots(gid_list, bbox_list=bbox_list) >>> bbox_list2 = ibs.get_annot_bboxes(aid_list) >>> vert_list2 = ibs.get_annot_verts(aid_list) >>> theta_list2 = ibs.get_annot_thetas(aid_list) >>> name_list2 = ibs.get_annot_names(aid_list) >>> print('Ensure=False. Should get back None chip fpaths') >>> chip_fpaths2 = ibs.get_annot_chip_fpath(aid_list, ensure=False) >>> assert [fpath is None for fpath in chip_fpaths2], 'should not have fpaths' >>> print('Ensure=True. Should get back None chip fpaths') >>> chip_fpaths = ibs.get_annot_chip_fpath(aid_list, ensure=True) >>> assert all([ut.checkpath(fpath, verbose=True) for fpath in chip_fpaths]), 'paths should exist' >>> ut.assert_eq(len(aid_list), num_add) >>> ut.assert_eq(len(vert_list2[0]), 4) >>> assert bbox_list2 == bbox_list, 'bboxes are unequal' >>> # Be sure to remove test annotation >>> # if this test fails a resetdbs might be nessary >>> result = '' >>> visual_uuid_list = ibs.get_annot_visual_uuids(aid_list) >>> semantic_uuid_list = ibs.get_annot_semantic_uuids(aid_list) >>> result += str(visual_uuid_list) + '\n' >>> result += str(semantic_uuid_list) + '\n' >>> print('Cleaning up. Removing added annotations') >>> ibs.delete_annots(aid_list) >>> assert not any([ut.checkpath(fpath, verbose=True) for fpath in chip_fpaths]), 'chip paths' >>> postvalid = ibs.get_valid_aids() >>> assert prevalid == postvalid, 'prevalid != postvalid' >>> result += str(postvalid) >>> print(result) [UUID('30f7639b-5161-a561-2c4f-41aed64e5b65'), UUID('5ccbb26d-104f-e655-cf2b-cf92e0ad2fd2')] [UUID('58905a72-dd31-c42b-d5b5-2312adfc7cba'), UUID('dd58665a-2a8b-8e84-4919-038c80bd9be0')] [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13] Example: >>> # Test with prevent_visual_duplicates on >>> # ENABLE_DOCTEST >>> from wbia.control.IBEISControl import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> prevalid = ibs.get_valid_aids() >>> num_add = 1 >>> gid_list = ibs.get_valid_gids()[0:1] * num_add >>> bbox_list = [(int(w * .1), int(h * .6), int(w * .5), int(h * .3)) ... for (w, h) in ibs.get_image_sizes(gid_list)] >>> bbox_list2 = [(int(w * .2), int(h * .6), int(w * .5), int(h * .3)) ... for (w, h) in ibs.get_image_sizes(gid_list)] >>> # Add a test annotation >>> print('Testing add_annots') >>> aid_list1 = ibs.add_annots(gid_list, bbox_list=bbox_list, prevent_visual_duplicates=True) >>> aid_list2 = ibs.add_annots(gid_list, bbox_list=bbox_list, prevent_visual_duplicates=True) >>> aid_list3 = ibs.add_annots(gid_list, bbox_list=bbox_list2, prevent_visual_duplicates=True) >>> assert aid_list1 == aid_list2, 'aid_list1 == aid_list2' >>> assert aid_list1 != aid_list3, 'aid_list1 != aid_list3' >>> aid_list_new = aid_list1 + aid_list3 >>> result = aid_list_new >>> print('Cleaning up. Removing added annotations') >>> ibs.delete_annots(aid_list_new) >>> print(result) [14, 15] """ assert yaw_list is None, 'yaw is depricated' if ut.VERBOSE: logger.info('[ibs] adding annotations') ut.assert_all_not_None(gid_list, 'gid_list') if len(gid_list) == 0: # nothing is being added logger.info('[ibs] WARNING: 0 annotations are beign added!') logger.info(ut.repr2(locals())) return [] preprocess_dict = ibs.compute_annot_visual_semantic_uuids( gid_list, include_preprocess=True, bbox_list=bbox_list, theta_list=theta_list, vert_list=vert_list, species_list=species_list, species_rowid_list=species_rowid_list, viewpoint_list=viewpoint_list, nid_list=nid_list, name_list=name_list, skip_cleaning=skip_cleaning, ) bbox_list = preprocess_dict['bbox_list'] theta_list = preprocess_dict['theta_list'] vert_list = preprocess_dict['vert_list'] species_rowid_list = preprocess_dict['species_rowid_list'] viewpoint_list = preprocess_dict['viewpoint_list'] viewpoint_ints = preprocess_dict['viewpoint_ints'] nid_list = preprocess_dict['nid_list'] # Build ~~deterministic?~~ random and unique ANNOTATION ids if annot_uuid_list is None: annot_uuid_list = [uuid.uuid4() for _ in range(len(gid_list))] # FIXME # Careful this code is very fragile. It might go out of sync # with the updating of the determenistic uuids. Find a way to # integrate both pieces of code without too much reundancy. # Make sure these tuples are constructed correctly if annot_visual_uuid_list is None: annot_visual_uuid_list = preprocess_dict['visual_uuid_list'] if annot_semantic_uuid_list is None: annot_semantic_uuid_list = preprocess_dict['semantic_uuid_list'] if detect_confidence_list is None: detect_confidence_list = [0.0 for _ in range(len(gid_list))] if notes_list is None: notes_list = ['' for _ in range(len(gid_list))] if quality_list is None: quality_list = [None] * len(gid_list) if multiple_list is None: multiple_list = [False] * len(gid_list) if interest_list is None: interest_list = [False] * len(gid_list) if canonical_list is None: canonical_list = [False] * len(gid_list) nVert_list = [len(verts) for verts in vert_list] vertstr_list = [str(verts) for verts in vert_list] xtl_list, ytl_list, width_list, height_list = list(zip(*bbox_list)) assert len(nVert_list) == len(vertstr_list) if staged_uuid_list is None: staged_uuid_list = [None] * len(gid_list) is_staged_list = [staged_uuid is not None for staged_uuid in staged_uuid_list] if staged_user_id_list is None: staged_user_id_list = [None] * len(gid_list) # Define arguments to insert props = ut.odict( [ ('annot_uuid', annot_uuid_list), ('image_rowid', gid_list), ('annot_xtl', xtl_list), ('annot_ytl', ytl_list), ('annot_width', width_list), ('annot_height', height_list), ('annot_theta', theta_list), ('annot_num_verts', nVert_list), ('annot_verts', vertstr_list), ('annot_viewpoint', viewpoint_list), ('annot_quality', quality_list), ('annot_toggle_multiple', multiple_list), ('annot_toggle_interest', interest_list), ('annot_toggle_canonical', canonical_list), ('annot_detect_confidence', detect_confidence_list), ('annot_note', notes_list), ('name_rowid', nid_list), ('species_rowid', species_rowid_list), ('annot_viewpoint_int', viewpoint_ints), ('annot_visual_uuid', annot_visual_uuid_list), ('annot_semantic_uuid', annot_semantic_uuid_list), ('annot_staged_flag', is_staged_list), ('annot_staged_uuid', staged_uuid_list), ('annot_staged_user_identity', staged_user_id_list), ] ) check_uuid_flags = [not isinstance(auuid, uuid.UUID) for auuid in annot_uuid_list] if any(check_uuid_flags): pos = ut.list_where(check_uuid_flags) raise ValueError('positions %r have malformated UUIDS' % (pos,)) colnames = tuple(props.keys()) params_iter = zip(*props.values()) # Execute add ANNOTATIONs SQL if prevent_visual_duplicates: # superkey_paramx = (len(colnames) - 2,) superkey_paramx = (colnames.index('annot_visual_uuid'),) get_rowid_from_superkey = ibs.get_annot_aids_from_visual_uuid else: # superkey_paramx = (0,) superkey_paramx = (colnames.index('annot_uuid'),) get_rowid_from_superkey = ibs.get_annot_aids_from_uuid aid_list = ibs.db.add_cleanly( const.ANNOTATION_TABLE, colnames, params_iter, get_rowid_from_superkey, superkey_paramx, ) # Updates both semantic and visual uuids # WE NEED TO DO THIS BEFOREHAND DUE TO VISUAL DUPLICATES # ibs.update_annot_visual_uuids(aid_list) # Invalidate image thumbnails, quiet_delete_thumbs causes no output on deletion from ut if delete_thumb: config2_ = {'thumbsize': 221} ibs.delete_image_thumbs(gid_list, quiet=quiet_delete_thumbs, **config2_) return aid_list
[docs]@register_ibs_method # @register_api('/api/annot/uuid/visual/info/', methods=['GET']) def get_annot_visual_uuid_info(ibs, aid_list): r""" Returns information used to compute annotation UUID. The image uuid, annotation verticies, are theta is hashted together to compute the visual uuid. The visual uuid does not include name or species information. get_annot_visual_uuid_info Args: aid_list (list): Returns: tuple: visual_infotup (image_uuid_list, verts_list, theta_list) SeeAlso: get_annot_visual_uuids get_annot_semantic_uuid_info CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_visual_uuid_info Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:2] >>> visual_infotup = ibs.get_annot_visual_uuid_info(aid_list) >>> result = str(list(zip(*visual_infotup))[0]) >>> print(result) (UUID('66ec193a-1619-b3b6-216d-1784b4833b61'), ((0, 0), (1047, 0), (1047, 715), (0, 715)), 0.0) """ # image_uuid_list = ibs.get_annot_image_uuids(aid_list) # verts_list = ibs.get_annot_verts(aid_list) # theta_list = ibs.get_annot_thetas(aid_list) # #visual_info_iter = zip(image_uuid_list, verts_list, theta_list, yaw_list) # #visual_info_list = list(visual_info_iter) # visual_infotup = (image_uuid_list, verts_list, theta_list) # Use predefined attributes annots = ibs.annots(aid_list) visual_infotup = tuple(getattr(annots, key) for key in VISUAL_TUP_PROPS) return visual_infotup
[docs]@register_ibs_method # @register_api('/api/annot/uuid/semantic/info/', methods=['GET']) def get_annot_semantic_uuid_info(ibs, aid_list, _visual_infotup=None): r""" Semenatic uuids are made up of visual and semantic information. Semantic information is name, species, yaw. Visual info is image uuid, verts, and theta Args: aid_list (list): _visual_infotup (tuple) : internal use only Returns: tuple: semantic_infotup (image_uuid_list, verts_list, theta_list, yaw_list, name_list, species_list) CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_semantic_uuid_info Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:2] >>> semantic_infotup = ibs.get_annot_semantic_uuid_info(aid_list) >>> result = ut.repr2(list(zip(*semantic_infotup))[1]) >>> print(result) (UUID('d8903434-942f-e0f5-d6c2-0dcbe3137bf7'), ((0, 0), (1035, 0), (1035, 576), (0, 576)), 0.0, 'left', 'easy', 'zebra_plains') """ # Semantic info depends on visual info if _visual_infotup is None: visual_infotup = get_annot_visual_uuid_info(ibs, aid_list) else: visual_infotup = _visual_infotup # Use predefined attributes annots = ibs.annots(aid_list) other_part = tuple(getattr(annots, key) for key in SEMANTIC_TUP_PROPS) # image_uuid_list, verts_list, theta_list = visual_infotup # It is visual info augmented with name and species # name_list = ibs.get_annot_names(aid_list) # species_list = ibs.get_annot_species_texts(aid_list) # viewpoint_codes = ibs.get_annot_viewpoint_code(aid_list) semantic_infotup = visual_infotup + other_part # (image_uuid_list, verts_list, theta_list, viewpoint_codes, name_list, # species_list) return semantic_infotup
[docs]@register_ibs_method @accessor_decors.cache_invalidator( const.ANNOTATION_TABLE, [ANNOT_SEMANTIC_UUID], rowidx=0 ) # @register_api('/api/annot/uuid/semantic/', methods=['PUT']) def update_annot_semantic_uuids(ibs, aid_list, _visual_infotup=None): r""" Ensures that annots have the proper semantic uuids """ semantic_infotup = ibs.get_annot_semantic_uuid_info(aid_list, _visual_infotup) assert len(semantic_infotup) == 6, 'len=%r' % (len(semantic_infotup),) annot_semantic_uuid_list = [ut.augment_uuid(*tup) for tup in zip(*semantic_infotup)] ibs.db.set( const.ANNOTATION_TABLE, (ANNOT_SEMANTIC_UUID,), annot_semantic_uuid_list, aid_list )
[docs]@register_ibs_method @accessor_decors.cache_invalidator( const.ANNOTATION_TABLE, [ANNOT_VISUAL_UUID, ANNOT_SEMANTIC_UUID], rowidx=0 ) # @register_api('/api/annot/uuid/visual/', methods=['PUT']) def update_annot_visual_uuids(ibs, aid_list): r""" Ensures that annots have the proper visual uuids Args: ibs (IBEISController): wbia controller object aid_list (list): list of annotation rowids CommandLine: python -m wbia.control.manual_annot_funcs update_annot_visual_uuids --db PZ_Master1 python -m wbia.control.manual_annot_funcs update_annot_visual_uuids python -m wbia update_annot_visual_uuids --db PZ_Master1 python -m wbia update_annot_visual_uuids --db PZ_Master0 python -m wbia update_annot_visual_uuids --db PZ_MTEST Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ibs._get_all_aids()[0:1] >>> update_annot_visual_uuids(ibs, aid_list) >>> result = ibs.get_annot_visual_uuids(aid_list)[0] >>> print(result) 8687dcb6-1f1f-fdd3-8b72-8f36f9f41905 """ visual_infotup = ibs.get_annot_visual_uuid_info(aid_list) logger.info('visual_infotup = %r' % (visual_infotup,)) assert len(visual_infotup) == 3, 'len=%r' % (len(visual_infotup),) annot_visual_uuid_list = [ut.augment_uuid(*tup) for tup in zip(*visual_infotup)] ibs.db.set( const.ANNOTATION_TABLE, (ANNOT_VISUAL_UUID,), annot_visual_uuid_list, aid_list ) # If visual uuids are changes semantic ones are also changed ibs.update_annot_semantic_uuids(aid_list, _visual_infotup=visual_infotup)
# ========== # IDERS # ========== # TODO CACHE THIS AND FIND WHAT IT SHOULD INVALIDATE IT # ADD ANNOTS, DELETE ANNOTS ANYTHING ELSE? @register_ibs_method @accessor_decors.ider def _get_all_aids(ibs): r""" Returns: list_ (list): all unfiltered aids (annotation rowids) """ all_aids = ibs.db.get_all_rowids(const.ANNOTATION_TABLE) return all_aids
[docs]@register_ibs_method def get_num_annotations(ibs, **kwargs): r""" Number of valid annotations """ aid_list = ibs.get_valid_aids(**kwargs) return len(aid_list)
[docs]@register_ibs_method @accessor_decors.ider @register_api('/api/annot/', methods=['GET']) def get_valid_aids( ibs, imgsetid=None, include_only_gid_list=None, yaw='no-filter', is_exemplar=None, is_staged=False, species=None, is_known=None, hasgt=None, minqual=None, has_timestamp=None, min_timedelta=None, ): r""" High level function for getting all annotation ids according a set of filters. Note: The yaw value cannot be None as a default because None is used as a filtering value Args: ibs (IBEISController): wbia controller object imgsetid (int): imageset id (default = None) include_only_gid_list (list): if specified filters annots not in these gids (default = None) yaw (str): (default = 'no-filter') is_exemplar (bool): if specified filters annots to either be or not be exemplars (default = None) species (str): (default = None) is_known (bool): (default = None) min_timedelta (int): minimum timedelta between annots of known individuals hasgt (bool): (default = None) Returns: list: aid_list - a list of valid ANNOTATION unique ids CommandLine: python -m wbia.control.manual_annot_funcs --test-get_valid_aids Ignore: ibs.print_annotation_table() RESTful: Method: GET URL: /api/annot/ Example: >>> # FIXME failing-test (22-Jul-2020) This test is failing and it's not clear how to fix it >>> # xdoctest: +SKIP >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> ut.exec_funckw(get_valid_aids, globals()) >>> imgsetid = 1 >>> yaw = 'no-filter' >>> species = ibs.const.TEST_SPECIES.ZEB_PLAIN >>> is_known = False >>> ibs.delete_all_imagesets() >>> ibs.compute_occurrences(config={'use_gps': False, 'seconds_thresh': 600}) >>> aid_list = get_valid_aids(ibs, imgsetid=imgsetid, species=species, is_known=is_known) >>> ut.assert_eq(ibs.get_annot_names(aid_list), [ibs.const.UNKNOWN] * 2, 'bad name') >>> ut.assert_eq(ibs.get_annot_species(aid_list), [species] * 2, 'bad species') >>> ut.assert_eq(ibs.get_annot_exemplar_flags(aid_list), [False] * 2, 'bad exemplar') >>> result = str(aid_list) >>> print(result) [1, 4] Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list1 = get_valid_aids(ibs, is_exemplar=True) >>> aid_list2 = get_valid_aids(ibs, is_exemplar=False) >>> intersect_aids = set(aid_list1).intersection(aid_list2) >>> ut.assert_eq(len(aid_list1), 9) >>> ut.assert_eq(len(aid_list2), 4) >>> ut.assert_eq(len(intersect_aids), 0) Ignore: import utool as ut setup = ut.codeblock( ''' import wbia ibs = wbia.opendb('PZ_Master1') ''' ) stmt_list = [ ut.codeblock( ''' ibs.db.get_all_rowids_where(ibs.const.ANNOTATION_TABLE, wbia.control.DB_SCHEMA.ANNOT_PARENT_ROWID + " IS NULL", tuple()) '''), ut.codeblock( ''' ibs.db.get_all_rowids(ibs.const.ANNOTATION_TABLE) '''), ] iterations = 100 verbose = True _ = ut.timeit_compare(stmt_list, setup=setup, iterations=iterations, verbose=verbose) """ # exemplar "imageset" (image group) if imgsetid is None: aid_list = ibs._get_all_aids() else: imagesettext = ibs.get_imageset_text(imgsetid) if imagesettext == const.EXEMPLAR_IMAGESETTEXT: is_exemplar = True aid_list = ibs.get_imageset_aids(imgsetid) aid_list = ibs.filter_annotation_set( aid_list, include_only_gid_list=include_only_gid_list, yaw=yaw, is_exemplar=is_exemplar, is_staged=is_staged, species=species, is_known=is_known, hasgt=hasgt, minqual=minqual, has_timestamp=has_timestamp, min_timedelta=min_timedelta, ) return aid_list
[docs]@register_ibs_method @register_api('/api/annot/<rowid>/', methods=['GET']) def annotation_src_api(rowid=None): r""" Returns the base64 encoded image of annotation <aid> RESTful: Method: GET URL: /api/annot/<aid>/ """ return routes_ajax.annotation_src(rowid)
[docs]@register_ibs_method def filter_annotation_set( ibs, aid_list, include_only_gid_list=None, yaw='no-filter', is_exemplar=None, is_staged=False, species=None, is_known=None, hasgt=None, minqual=None, has_timestamp=None, sort=False, is_canonical=None, min_timedelta=None, ): if not aid_list: # no need to filter if empty return aid_list # -- valid aid filtering -- # filter by is_exemplar if is_exemplar is True: # corresponding unoptimized hack for is_exemplar flag_list = ibs.get_annot_exemplar_flags(aid_list) aid_list = ut.compress(aid_list, flag_list) elif is_exemplar is False: flag_list = ibs.get_annot_exemplar_flags(aid_list) aid_list = ut.filterfalse_items(aid_list, flag_list) # filter by is_staged if is_staged is True: # corresponding unoptimized hack for is_staged flag_list = ibs.get_annot_staged_flags(aid_list) aid_list = ut.compress(aid_list, flag_list) elif is_staged is False: flag_list = ibs.get_annot_staged_flags(aid_list) aid_list = ut.filterfalse_items(aid_list, flag_list) # filter by is_canonical if is_canonical is True: flag_list = ibs.get_annot_canonical(aid_list) aid_list = ut.compress(aid_list, flag_list) elif is_canonical is False: flag_list = ibs.get_annot_canonical(aid_list) aid_list = ut.filterfalse_items(aid_list, flag_list) if include_only_gid_list is not None: gid_list = ibs.get_annot_gids(aid_list) is_valid_gid = [gid in include_only_gid_list for gid in gid_list] aid_list = ut.compress(aid_list, is_valid_gid) if yaw != 'no-filter': yaw_list = ibs.get_annot_yaws(aid_list) is_valid_yaw = [yaw == flag for flag in yaw_list] aid_list = ut.compress(aid_list, is_valid_yaw) if species is not None: aid_list = ibs.filter_aids_to_species(aid_list, species) if is_known is not None: aid_list = ibs.filter_aids_without_name(aid_list, invert=not is_known) if minqual is not None: aid_list = ibs.filter_aids_to_quality(aid_list, minqual, unknown_ok=True) if has_timestamp is not None: aid_list = ibs.filter_aids_without_timestamps(aid_list, invert=not has_timestamp) if min_timedelta is not None: aid_list = ibs.filter_annots_using_minimum_timedelta(aid_list, min_timedelta) if hasgt: hasgt_list = ibs.get_annot_has_groundtruth(aid_list) aid_list = ut.compress(aid_list, hasgt_list) if sort: aid_list = sorted(aid_list) return aid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_aid(ibs, aid_list, eager=True, nInput=None): """self verifier Example: >>> # ENABLE_DOCTEST >>> from wbia.control.IBEISControl import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() + [None, -1, 10434320432] >>> aid_list_ = ibs.get_annot_aid(aid_list) >>> assert [r is None for r in aid_list_[-3:]] >>> assert [r is not None for r in aid_list_[0:-3]] """ id_iter = aid_list colnames = (ANNOT_ROWID,) aid_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid', eager=eager, nInput=nInput, ) return aid_list
[docs]@register_ibs_method # @register_api('/api/annot/rows/', methods=['GET']) def get_annot_rows(ibs, aid_list): r""" Auto-docstr for 'get_annot_rows' """ colnames = ( 'annot_uuid', 'image_rowid', 'annot_xtl', 'annot_ytl', 'annot_width', 'annot_height', 'annot_theta', 'annot_num_verts', 'annot_verts', ANNOT_YAW, 'annot_detect_confidence', 'annot_note', 'name_rowid', 'species_rowid', 'annot_visual_uuid', 'annot_semantic_uuid', ) rows_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, aid_list, unpack_scalars=False ) return rows_list
# ========== # DELETERS # ==========
[docs]@register_ibs_method @accessor_decors.deleter @register_api('/api/annot/name/rowid/', methods=['DELETE']) def delete_annot_nids(ibs, aid_list): r""" Remove name assocation from the list of input aids. Does this by setting each annotations nid to the UNKNOWN name rowid RESTful: Method: DELETE URL: /api/annot/name/rowid/ """ # FIXME: This should be implicit by setting the anotation name to the # unknown name # ibs.delete_annot_relations_oftype(aid_list, const.INDIVIDUAL_KEY) ibs.set_annot_name_rowids(aid_list, [const.UNKNOWN_NAME_ROWID] * len(aid_list))
[docs]@register_ibs_method @accessor_decors.deleter @register_api('/api/annot/species/rowid/', methods=['DELETE'], __api_plural_check__=False) def delete_annot_speciesids(ibs, aid_list): r""" Deletes nids of a list of annotations RESTful: Method: DELETE URL: /api/annot/species/rowid/ """ # FIXME: This should be implicit by setting the anotation name to the # unknown species # ibs.delete_annot_relations_oftype(aid_list, const.SPECIES_KEY) ibs.set_annot_species_rowids(aid_list, [const.UNKNOWN_SPECIES_ROWID] * len(aid_list))
[docs]@register_ibs_method @accessor_decors.deleter @accessor_decors.cache_invalidator(const.ANNOTATION_TABLE, rowidx=0) @accessor_decors.cache_invalidator( const.IMAGE_TABLE, colnames=[ANNOT_ROWIDS], rowidx=None ) @accessor_decors.cache_invalidator(const.NAME_TABLE, colnames=[ANNOT_ROWIDS], rowidx=None) @accessor_decors.cache_invalidator( const.IMAGESET_TABLE, ['percent_names_with_exemplar_str'] ) @register_api('/api/annot/', methods=['DELETE']) def delete_annots(ibs, aid_list): r""" deletes annotations from the database RESTful: Method: DELETE URL: /api/annot/ Args: ibs (IBEISController): wbia controller object aid_list (int): list of annotation ids CommandLine: python -m wbia.control.manual_annot_funcs --test-delete_annots python -m wbia.control.manual_annot_funcs --test-delete_annots --debug-api-cache python -m wbia.control.manual_annot_funcs --test-delete_annots SeeAlso: back.delete_annot Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> from os.path import exists >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> ibs.delete_empty_nids() >>> # Add some annotations to delete >>> num_add = 2 >>> gid_list = ibs.get_valid_gids()[0:num_add] >>> nid = ibs.make_next_nids(1)[0] >>> nid_list = [nid] * num_add >>> bbox_list = [(int(w * .1), int(h * .6), int(w * .5), int(h * .3)) ... for (w, h) in ibs.get_image_sizes(gid_list)] >>> new_aid_list = ibs.add_annots(gid_list, bbox_list=bbox_list, >>> nid_list=nid_list) >>> ibs.get_annot_nids(new_aid_list) >>> ut.assert_lists_eq(ibs.get_annot_nids(new_aid_list), nid_list) >>> assert ibs.get_name_aids(nid) == new_aid_list, 'annots should all have same name' >>> assert new_aid_list == ibs.get_name_aids(nid), 'inverse name mapping should work' >>> #thumpaths = ibs.get_image_thumbpath(gid_list, ensure_paths=True, **{'thumbsize': 221}) >>> #assert any(ut.lmap(exists, thumpaths)), 'thumbs should be there' >>> before_aids = ibs.get_image_aids(gid_list) >>> print('BEFORE gids: ' + str(before_aids)) >>> result = ibs.delete_annots(new_aid_list) >>> assert ibs.get_name_aids(nid) == [], 'annots should be removed' >>> after_aids = ibs.get_image_aids(gid_list) >>> #thumpaths = ibs.get_image_thumbpath(gid_list, ensure_paths=False, **{'thumbsize': 221}) >>> #assert not any(ut.lmap(exists, thumpaths)), 'thumbs should be gone' >>> assert after_aids != before_aids, 'the invalidators must have bugs' >>> print('AFTER gids: ' + str(after_aids)) >>> valid_aids = ibs.get_valid_aids() >>> assert [aid not in valid_aids for aid in new_aid_list], 'should no longer be valid aids' >>> print(result) >>> ibs.delete_empty_nids() """ if ut.VERBOSE: logger.info('[ibs] deleting %d annotations' % len(aid_list)) # FIXME: Need to reliabely delete thumbnails # config2_ = {'draw_annots': True, 'thumbsize': 221} # MEGA HACK FOR QT ibs.delete_annot_imgthumbs(aid_list) # Delete chips and features first # ibs.delete_annot_relations(aid_list) ibs.delete_annot_chips(aid_list) ibs.depc_annot.delete_root(aid_list) # TODO: # delete parent rowid column if exists in annot table return ibs.db.delete_rowids(const.ANNOTATION_TABLE, aid_list)
[docs]@register_ibs_method @accessor_decors.deleter def delete_annot_imgthumbs(ibs, aid_list): gid_list_ = ibs.get_annot_gids(aid_list) ibs.delete_image_thumbs(gid_list_)
# # Less hacky # table_config_filter = { # 'thumbnails': { # 'draw_annots': True, # } # } # ibs.depc_image.delete_root(gid_list_, table_config_filter) # ibs.delete_image_thumbs(gid_list_) # # MEGA HACK FOR QT # config2_ = {'thumbsize': 221} # ibs.delete_image_thumbs(gid_list_, **config2_) # # ibs.delete_image_thumbs(gid_list_) # ========== # GETTERS # ==========
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/aids/uuid/semantic/', methods=['GET']) def get_annot_aids_from_semantic_uuid(ibs, semantic_uuid_list): r""" Args: semantic_uuid_list (list): Returns: list: annot rowids """ aids_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_ROWID,), semantic_uuid_list, id_colname=ANNOT_SEMANTIC_UUID, ) return aids_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/rowid/uuid/', methods=['GET']) def get_annot_aids_from_uuid(ibs, uuid_list): r""" Returns: list_ (list): annot rowids RESTful: Method: GET URL: /api/annot/rowid/uuid/ """ # FIXME: MAKE SQL-METHOD FOR NON-ROWID GETTERS aids_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_ROWID,), uuid_list, id_colname=ANNOT_UUID ) return aids_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/uuid/missing/', methods=['GET']) def get_annot_missing_uuid(ibs, uuid_list): r""" Returns: list_ (list): a list of missing annot uuids """ aid_list = ibs.get_annot_aids_from_uuid(uuid_list) zipped = zip(aid_list, uuid_list) missing_uuid_list = [uuid for aid, uuid in zipped if aid is None] return missing_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/aids/uuid/visual/', methods=['GET']) def get_annot_aids_from_visual_uuid(ibs, visual_uuid_list): r""" Args: visual_uuid_list (list): Returns: list: annot rowids """ aids_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_ROWID,), visual_uuid_list, id_colname=ANNOT_VISUAL_UUID, ) return aids_list
get_annot_rowids_from_visual_uuid = get_annot_aids_from_visual_uuid
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1toM @register_api('/api/annot/bbox/', methods=['GET']) def get_annot_bboxes(ibs, aid_list): r""" Returns: bbox_list (list): annotation bounding boxes in image space RESTful: Method: GET URL: /api/annot/bbox/ """ colnames = ( 'annot_xtl', 'annot_ytl', 'annot_width', 'annot_height', ) bbox_list = ibs.db.get(const.ANNOTATION_TABLE, colnames, aid_list) return bbox_list
[docs]@register_ibs_method # @register_api('/api/annot/labels/', methods=['GET']) def get_annot_class_labels(ibs, aid_list): r""" DEPRICATE? Returns: list of tuples: identifying animal name and view """ name_list = ibs.get_annot_name_rowids(aid_list) # TODO: use yaw? yaw_list = [0 for _ in name_list] classlabel_list = list(zip(name_list, yaw_list)) return classlabel_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/detect/confidence/', methods=['GET']) def get_annot_detect_confidence(ibs, aid_list): r""" Returns: list_ (list): a list confidences that the annotations is a valid detection RESTful: Method: GET URL: /api/annot/detect/confidence/ """ annot_detect_confidence_list = ibs.db.get( const.ANNOTATION_TABLE, ('annot_detect_confidence',), aid_list ) return annot_detect_confidence_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/exemplar/', methods=['GET']) def get_annot_exemplar_flags(ibs, aid_list): r""" returns if an annotation is an exemplar Args: ibs (IBEISController): wbia controller object aid_list (int): list of annotation ids Returns: list: annot_exemplar_flag_list - True if annotation is an exemplar CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_exemplar_flags RESTful: Method: GET URL: /api/annot/exemplar/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> gid_list = get_annot_exemplar_flags(ibs, aid_list) >>> result = str(gid_list) >>> print(result) """ annot_exemplar_flag_list = ibs.db.get( const.ANNOTATION_TABLE, ('annot_exemplar_flag',), aid_list ) return annot_exemplar_flag_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 # @cache_getter(const.ANNOTATION_TABLE, 'image_rowid') @register_api('/api/annot/image/rowid/', methods=['GET']) def get_annot_gids(ibs, aid_list, assume_unique=False): r""" Get parent image rowids of annotations Args: aid_list (list): Returns: gid_list (list): image rowids RESTful: Method: GET URL: /api/annot/image/rowid/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> result = get_annot_gids(ibs, aid_list) >>> print(result) """ gid_list = ibs.db.get( const.ANNOTATION_TABLE, ('image_rowid',), aid_list, assume_unique=assume_unique ) return gid_list
[docs]@register_ibs_method def get_annot_image_rowids(ibs, aid_list): return ibs.get_annot_gids(aid_list)
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 # @cache_getter(const.ANNOTATION_TABLE, 'image_rowid') @register_api('/api/annot/imageset/rowid/', methods=['GET']) def get_annot_imgsetids(ibs, aid_list): r""" Get parent image rowids of annotations Args: aid_list (list): Returns: imgsetid_list (list): imageset rowids RESTful: Method: GET URL: /api/annot/imageset/rowid/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> result = get_annot_gids(ibs, aid_list) >>> print(result) """ gid_list = ibs.db.get(const.ANNOTATION_TABLE, ('image_rowid',), aid_list) imgsetids_list = ibs.get_image_imgsetids(gid_list) return imgsetids_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 # @cache_getter(const.ANNOTATION_TABLE, 'image_rowid') @register_api('/api/annot/imageset/uuid/', methods=['GET']) def get_annot_imgset_uuids(ibs, aid_list): r""" Get parent image rowids of annotations Args: aid_list (list): Returns: imgset_uuid_list (list): imageset uuids RESTful: Method: GET URL: /api/annot/imageset/uuid/ """ imgsetids_list = ibs.get_annot_imgsetids(aid_list) imgset_uuid_list = ibs.get_imageset_uuid(imgsetids_list) return imgset_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1toM # @register_api('/api/annot/gar/rowid/', methods=['GET']) def get_annot_gar_rowids(ibs, aid_list): r""" Auto-docstr for 'get_annot_gar_rowids' """ colnames = (GAR_ROWID,) gar_rowid_list = ibs.db.get( const.GA_RELATION_TABLE, colnames, aid_list, id_colname=ANNOT_ROWID, unpack_scalars=False, ) return gar_rowid_list
[docs]@register_ibs_method @accessor_decors.getter_1toM # @register_api('/api/annot/aids/otherimage/', methods=['GET']) def get_annot_otherimage_aids(ibs, aid_list, daid_list=None, assume_unique=False): r""" Auto-docstr for 'get_annot_otherimage_aids' """ gid_list = ibs.get_annot_gids(aid_list, assume_unique=assume_unique) if daid_list is None: image_aids_list = ibs.get_image_aids(gid_list) # Remove self from list other_aids_list = [ list(set(aids) - {aid}) for aids, aid in zip(image_aids_list, aid_list) ] else: daids = np.array(daid_list) internal_data_gids = ibs.get_annot_gids(daids, assume_unique=assume_unique) other_aids_list = [daids.compress(internal_data_gids == gid) for gid in gid_list] return other_aids_list
[docs]@register_ibs_method @accessor_decors.getter_1toM # @register_api('/api/annot/aids/contact/', methods=['GET']) def get_annot_contact_aids( ibs, aid_list, daid_list=None, check_isect=False, assume_unique=False ): r""" Returns the other aids that appear in the same image that this annotation is from. Args: ibs (IBEISController): wbia controller object aid_list (list): CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_contact_aids;1 Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> contact_aids = ibs.get_annot_contact_aids(aid_list) >>> contact_gids = ibs.unflat_map(ibs.get_annot_gids, contact_aids) >>> gid_list = ibs.get_annot_gids(aid_list) >>> for gids, gid, aids, aid in zip(contact_gids, gid_list, contact_aids, aid_list): ... assert ut.allsame(gids), 'annots should be from same image' ... assert len(gids) == 0 or gids[0] == gid, 'and same image as parent annot' ... assert aid not in aids, 'should not include self' Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb2') >>> aid_list = ibs.get_valid_aids() >>> contact_aids = ibs.get_annot_contact_aids(aid_list) >>> contact_gids = ibs.unflat_map(ibs.get_annot_gids, contact_aids) >>> gid_list = ibs.get_annot_gids(aid_list) >>> print('contact_aids = %r' % (contact_aids,)) >>> for gids, gid, aids, aid in zip(contact_gids, gid_list, contact_aids, aid_list): ... assert ut.allsame(gids), 'annots should be from same image' ... assert len(gids) == 0 or gids[0] == gid, 'and same image as parent annot' ... assert aid not in aids, 'should not include self' """ other_aids_list = ibs.get_annot_otherimage_aids( aid_list, daid_list=daid_list, assume_unique=assume_unique ) if check_isect: import shapely.geometry # TODO: might not be accounting for rotated verticies verts_list = ibs.get_annot_verts(aid_list) other_verts_list = ibs.unflat_map(ibs.get_annot_verts, other_aids_list) poly_list = [shapely.geometry.Polygon(verts) for verts in verts_list] other_polys_list = [ [shapely.geometry.Polygon(verts) for verts in _] for _ in other_verts_list ] flags_list = [ [p1.intersects(p2) for p2 in p2_list] for p1, p2_list in zip(poly_list, other_polys_list) ] contact_aids = [ ut.compress(other_aids, flags) for other_aids, flags in zip(other_aids_list, flags_list) ] else: contact_aids = other_aids_list return contact_aids
# @register_ibs_method # @accessor_decors.getter_1toM # def get_annot_intersecting_aids(ibs, aid_list): # pass
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_num_contact_aids(ibs, aid_list): r""" Auto-docstr for 'get_annot_num_contact_aids' """ nOther_aids_list = list(map(len, ibs.get_annot_contact_aids(aid_list))) return nOther_aids_list
[docs]@register_ibs_method @accessor_decors.getter_1toM def get_annot_groundfalse( ibs, aid_list, valid_aids=None, filter_unknowns=True, daid_list=None ): r""" gets all annotations with different names Returns: groundfalse_list (list): a list of aids which are known to be different for each Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> groundfalse_list = get_annot_groundfalse(ibs, aid_list) >>> result = str(groundfalse_list) >>> print(result) """ if valid_aids is None: # get all valid aids if not specified valid_aids = ibs.get_valid_aids() if daid_list is not None: valid_aids = list(set(daid_list).intersection(set(valid_aids))) if filter_unknowns: # Remove aids which do not have a name isunknown_list = ibs.is_aid_unknown(valid_aids) valid_aids_ = ut.filterfalse_items(valid_aids, isunknown_list) else: valid_aids_ = valid_aids # Build the set of groundfalse annotations # nid_list = ibs.get_annot_name_rowids(aid_list) # aids_list = ibs.get_name_aids(nid_list, enable_unknown_fix=True) # aids_setlist = map(set, aids_list) # valid_aids = set(valid_aids_) # groundfalse_list = [list(valid_aids - aids) for aids in aids_setlist] nid_list = ibs.get_annot_name_rowids(aid_list) nid_to_aids = ut.group_items(aid_list, nid_list) nid_to_aidset = ut.map_dict_vals(set, nid_to_aids) valid_aids = set(valid_aids_) groundfalse_list = [list(valid_aids - nid_to_aidset[nid]) for nid in nid_list] return groundfalse_list
[docs]@register_ibs_method @accessor_decors.getter_1toM # @register_api('/api/annot/groundtruth/', methods=['GET']) def get_annot_groundtruth(ibs, aid_list, is_exemplar=None, noself=True, daid_list=None): r""" gets all annotations with the same names Args: aid_list (list): list of annotation rowids to get groundtruth of is_exemplar (None): noself (bool): daid_list (list): Returns: groundtruth_list (list): a list of aids with the same name foreach aid in aid_list. a set of aids belonging to the same name is called a groundtruth. A list of these is called a groundtruth_list. CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_groundtruth:0 python -m wbia.control.manual_annot_funcs --test-get_annot_groundtruth:1 python -m wbia.control.manual_annot_funcs --test-get_annot_groundtruth:2 python -m --tf get_annot_groundtruth:0 --db=PZ_Master0 --aids=97 --exec-mode Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ut.get_argval('--aids', list, ibs.get_valid_aids()) >>> is_exemplar, noself, daid_list = None, True, None >>> groundtruth_list = ibs.get_annot_groundtruth(aid_list, is_exemplar, noself, daid_list) >>> result = 'groundtruth_list = ' + str(groundtruth_list) >>> print(result) groundtruth_list = [[], [3], [2], [], [6], [5], [], [], [], [], [], [], []] Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> is_exemplar, noself, daid_list = True, True, None >>> groundtruth_list = ibs.get_annot_groundtruth(aid_list, is_exemplar, noself, daid_list) >>> result = str(groundtruth_list) >>> print(result) [[], [3], [2], [], [6], [5], [], [], [], [], [], [], []] Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> is_exemplar, noself, daid_list = False, False, aid_list >>> groundtruth_list = ibs.get_annot_groundtruth(aid_list, is_exemplar, noself, daid_list) >>> result = str(groundtruth_list) >>> print(result) [[1], [], [], [4], [], [], [], [], [9], [], [11], [], []] """ # TODO: Optimize nid_list = ibs.get_annot_name_rowids(aid_list) # if daid_list is not None: # # when given a valid pool try to skip the get_name_aids call # aids_list_, nid_list_ = ibs.group_annots_by_name(daid_list, distinguish_unknowns=True) # nid2_aids = dict(zip(nid_list_, aids_list_)) # aids_list = ut.dict_take(nid2_aids, nid_list, []) # else: aids_list = ibs.get_name_aids(nid_list, enable_unknown_fix=True) if is_exemplar is None: groundtruth_list_ = aids_list else: # Filter out non-exemplars exemplar_flags_list = ibsfuncs.unflat_map(ibs.get_annot_exemplar_flags, aids_list) isvalids_list = [ [flag == is_exemplar for flag in flags] for flags in exemplar_flags_list ] groundtruth_list_ = [ ut.compress(aids, isvalids) for aids, isvalids in zip(aids_list, isvalids_list) ] if noself: # Remove yourself from the set groundtruth_list = [ list(set(aids) - {aid}) for aids, aid in zip(groundtruth_list_, aid_list) ] else: groundtruth_list = groundtruth_list_ if daid_list is not None: # filter out any groundtruth that isn't allowed daid_set = set(daid_list) groundtruth_list = [ list(daid_set.intersection(set(aids))) for aids in groundtruth_list ] return groundtruth_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/groundtruth/check/', methods=['GET']) def get_annot_has_groundtruth( ibs, aid_list, is_exemplar=None, noself=True, daid_list=None ): r""" Args: aid_list (list): is_exemplar (None): noself (bool): daid_list (list): Returns: list: has_gt_list CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_has_groundtruth Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> is_exemplar = None >>> noself = True >>> daid_list = None >>> has_gt_list = get_annot_has_groundtruth(ibs, aid_list, is_exemplar, noself, daid_list) >>> result = str(has_gt_list) >>> print(result) """ # TODO: Optimize numgts_list = ibs.get_annot_num_groundtruth( aid_list, is_exemplar=is_exemplar, noself=noself, daid_list=daid_list ) has_gt_list = [num_gts > 0 for num_gts in numgts_list] return has_gt_list
# @register_ibs_method # def get_annot_hashid_rowid(ibs, aid_list, prefix=''): # raise AssertionError('You probably want to call a uuid hash id method') # label = ''.join(('_', prefix, 'UUIDS')) # aids_hashid = ut.hashstr_arr(aid_list, label) # return aids_hashid
[docs]@register_ibs_method @register_api('/api/annot/uuid/hashid/', methods=['GET']) def get_annot_hashid_uuid(ibs, aid_list, prefix=''): r""" builds an aggregate random hash id for a list of aids RESTful: Method: GET URL: /api/annot/uuid/hashid/ """ uuid_list = ibs.get_annot_uuids(aid_list) label = ''.join(('_', prefix, 'UUIDS')) uuid_hashid = ut.hashstr_arr(uuid_list, label) return uuid_hashid
[docs]@register_ibs_method def get_annot_hashid_visual_uuid(ibs, aid_list, prefix='', pathsafe=False): r""" builds an aggregate visual hash id for a list of aids Args: _new (bool): Eventually we will change the hashing scheme and all old data will be invalidated. (default=False) """ visual_uuid_list = ibs.get_annot_visual_uuids(aid_list) label = ''.join(('_', prefix, 'VUUIDS')) visual_uuid_hashid = ut.hashstr_arr27(visual_uuid_list, label, pathsafe=pathsafe) # TODO: use this implementation instead # visual_uuid_list = ibs.get_annot_visual_uuids(aid_list) # label = ''.join((prefix, 'VUUIDS')) # visual_uuid_hashid = ut.hashid_arr(visual_uuid_list, label) return visual_uuid_hashid
[docs]@register_ibs_method def get_annot_hashid_semantic_uuid(ibs, aid_list, prefix=''): r""" builds an aggregate semantic hash id for a list of aids Args: ibs (wbia.IBEISController): wbia controller object aid_list (list): list of annotation rowids prefix (str): (default = '') _new (bool): Eventually we will change the hashing scheme and all old data will be invalidated. (default=False) Returns: str: semantic_uuid_hashid CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_hashid_semantic_uuid Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ibs.get_valid_aids() >>> annots = ibs.annots() >>> prefix = '' >>> semantic_uuid_hashid = get_annot_hashid_semantic_uuid(ibs, aid_list, prefix) >>> result = ut.repr2(annots.semantic_uuids[0:2], nl=1) + '\n' >>> result += ('semantic_uuid_hashid = ' + str(semantic_uuid_hashid)) >>> print(result) [ UUID('...'), UUID('...'), ] semantic_uuid_hashid = SUUIDS-13-... """ semantic_uuid_list = ibs.get_annot_semantic_uuids(aid_list) label = ''.join((prefix, 'SUUIDS')) semantic_uuid_hashid = ut.hashid_arr(semantic_uuid_list, label) return semantic_uuid_hashid
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/theta/', methods=['GET']) def get_annot_thetas(ibs, aid_list): r""" Returns: theta_list (list): a list of floats describing the angles of each chip CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_thetas RESTful: Method: GET URL: /api/annot/theta/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('NAUT_test') >>> aid_list = ibs.get_valid_aids() >>> result = get_annot_thetas(ibs, aid_list) >>> print(result) [2.75742, 0.792917, 2.53605, 2.67795, 0.946773, 2.56729] """ theta_list = ibs.db.get(const.ANNOTATION_TABLE, ('annot_theta',), aid_list) return theta_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/uuid/', methods=['GET']) def get_annot_uuids(ibs, aid_list): r""" Returns: list: annot_uuid_list a list of image uuids by aid RESTful: Method: GET URL: /api/annot/uuid/ """ annot_uuid_list = ibs.db.get(const.ANNOTATION_TABLE, ('annot_uuid',), aid_list) return annot_uuid_list
[docs]@register_ibs_method # @register_api('/api/annot/uuid/valid/', methods=['GET']) def get_valid_annot_uuids(ibs): r""" Returns: list: annot_uuid_list a list of image uuids for all valid aids """ aid_list = ibs.get_valid_aids() annot_uuid_list = ibs.get_annot_uuids(aid_list) return annot_uuid_list
# It is a good idea to have the cache on for the annot uuids, they are quite slow to load
[docs]@register_ibs_method @accessor_decors.getter_1to1 @accessor_decors.cache_getter(const.ANNOTATION_TABLE, ANNOT_SEMANTIC_UUID) # @register_api('/api/annot/uuid/semantic/', methods=['GET']) def get_annot_semantic_uuids(ibs, aid_list): r""" annot_semantic_uuid_list <- annot.annot_semantic_uuid[aid_list] gets data from the "native" column "annot_semantic_uuid" in the "annot" table Args: aid_list (list): Returns: list: annot_semantic_uuid_list CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_semantic_uuids Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> ibs, qreq_ = testdata_ibs() >>> aid_list = ibs._get_all_aids()[0:1] >>> annot_semantic_uuid_list = ibs.get_annot_semantic_uuids(aid_list) >>> assert len(aid_list) == len(annot_semantic_uuid_list) >>> print(annot_semantic_uuid_list) [UUID('9acc1a8e-b35f-11b5-f844-9e8fd5dd7ad9')] """ id_iter = aid_list colnames = (ANNOT_SEMANTIC_UUID,) annot_semantic_uuid_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid' ) return annot_semantic_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @accessor_decors.cache_getter(const.ANNOTATION_TABLE, ANNOT_VISUAL_UUID) # @register_api('/api/annot/uuid/visual/', methods=['GET']) def get_annot_visual_uuids(ibs, aid_list): r""" The image uuid, annotation verticies, are theta is hashted together to compute the visual uuid. The visual uuid does not include name or species information. annot_visual_uuid_list <- annot.annot_visual_uuid[aid_list] gets data from the "native" column "annot_visual_uuid" in the "annot" table Args: aid_list (list): Returns: list: annot_visual_uuid_list CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_visual_uuids Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> ibs, qreq_ = testdata_ibs() >>> aid_list = ibs._get_all_aids()[0:1] >>> annot_visual_uuid_list = ibs.get_annot_visual_uuids(aid_list) >>> assert len(aid_list) == len(annot_visual_uuid_list) >>> print(annot_visual_uuid_list) [UUID('8687dcb6-1f1f-fdd3-8b72-8f36f9f41905')] [UUID('76de0416-7c92-e1b3-4a17-25df32e9c2b4')] """ id_iter = aid_list colnames = (ANNOT_VISUAL_UUID,) annot_visual_uuid_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid' ) return annot_visual_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/vert/', methods=['GET']) def get_annot_verts(ibs, aid_list): r""" Returns: vert_list (list): the vertices that form the polygon of each chip RESTful: Method: GET URL: /api/annot/vert/ """ from wbia.algo.preproc import preproc_annot vertstr_list = ibs.db.get(const.ANNOTATION_TABLE, ('annot_verts',), aid_list) vert_list = preproc_annot.postget_annot_verts(vertstr_list) # vert_list = [eval(vertstr, {}, {}) for vertstr in vertstr_list] return vert_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/vert/rotated/', methods=['GET']) def get_annot_rotated_verts(ibs, aid_list): r""" Returns: rotated_vert_list (list): verticies after rotation by theta. RESTful: Method: GET URL: /api/annot/vert/rotated/ """ import vtool as vt vert_list = ibs.get_annot_verts(aid_list) theta_list = ibs.get_annot_thetas(aid_list) # Convex bounding boxes for verticies bbox_list = vt.geometry.bboxes_from_vert_list(vert_list) rot_list = [ vt.rotation_around_bbox_mat3x3(theta, bbox) for theta, bbox in zip(theta_list, bbox_list) ] rotated_vert_list = [ vt.transform_points_with_homography(rot, np.array(verts).T).T.tolist() for rot, verts in zip(rot_list, vert_list) ] # vert_list = [eval(vertstr, {}, {}) for vertstr in vertstr_list] return rotated_vert_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @accessor_decors.cache_getter(const.ANNOTATION_TABLE, ANNOT_YAW) @register_api('/api/annot/yaw/', methods=['GET']) # @profile def get_annot_yaws(ibs, aid_list, assume_unique=False): r""" A yaw is the yaw of the annotation in radians yaw is inverted. Will be fixed soon. DEPRICATE The following views have these angles of yaw: left side - 0.50 tau radians front side - 0.25 tau radians right side - 0.00 tau radians back side - 0.75 tau radians tau = 2 * pi SeeAlso: wbia.const.VIEWTEXT_TO_YAW_RADIANS Returns: yaw_list (list): the yaw (in radians) for the annotation CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_yaws RESTful: Method: GET URL: /api/annot/yaw/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[::3] >>> result = get_annot_yaws(ibs, aid_list) >>> print(result) [3.141592653589793, 3.141592653589793, None, 3.141592653589793, None] """ yaw_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_YAW,), aid_list, assume_unique=assume_unique ) # if False: # yaw_list2 = [ibs.db.cur.execute('SELECT annot_yaw from annotations WHERE rowid=?', (aid,)).fetchone()[0] for aid in aid_list] # # misses cases when aid_list is not unique? # yaw_list3 = ut.take_column(ibs.db.cur.execute('SELECT annot_yaw from annotations WHERE rowid IN (%s) ORDER BY rowid ASC' % (','.join(map(str, aid_list)),)).fetchall(), 0) # sortx = ut.argsort(ut.argsort(aid_list)) # yaw_list3 = ut.take(yaw_list3, sortx) # ut.make_index_lookup(aid_list) yaw_list = [yaw if yaw is not None and yaw >= 0.0 else None for yaw in yaw_list] return yaw_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/viewpoint/', methods=['GET']) def get_annot_viewpoints(ibs, aid_list, assume_unique=False): r""" Returns: viewpoint_text (list): the viewpoint for the annotation RESTful: Method: GET URL: /api/annot/viewpoint/ """ viewpoint_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_VIEWPOINT,), aid_list, assume_unique=assume_unique ) return viewpoint_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_yaws_asfloat(ibs, aid_list): r""" Ensures that Nones are returned as nans DEPRICATE """ yaw_list = ibs.get_annot_yaws(aid_list) yaw_list = np.array(ut.replace_nones(yaw_list, np.nan)) return yaw_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/yaw/text/', methods=['GET']) def get_annot_yaw_texts(ibs, aid_list, assume_unique=False): r""" Auto-docstr for 'get_annot_yaw_texts' DEPRICATE RESTful: Method: GET URL: /api/annot/yaw/text/ """ yaw_list = ibs.get_annot_yaws(aid_list, assume_unique=assume_unique) yaw_text_list = ibsfuncs.get_yaw_viewtexts(yaw_list) return yaw_text_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_viewpoint_int(ibs, aids, assume_unique=False): # Pre 1.7.0 # yaw_texts = ibs.get_annot_yaw_texts(aid_list, assume_unique=assume_unique) # VIEW = ibs.const.VIEW # UNKNOWN_CODE = VIEW.INT_TO_CODE[VIEW.UNKNOWN] # yaw_texts2 = (UNKNOWN_CODE if y is None else y for y in yaw_texts) # view_ints = ut.dict_take(ibs.const.VIEW.CODE_TO_INT, yaw_texts2) # return view_ints # --------- # Post 1.7.0 return ibs.db.get( const.ANNOTATION_TABLE, ('annot_viewpoint_int',), aids, assume_unique=assume_unique, )
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/staged/', methods=['GET']) def get_annot_staged_flags(ibs, aid_list): r""" returns if an annotation is staged Args: ibs (IBEISController): wbia controller object aid_list (int): list of annotation ids Returns: list: annot_staged_flag_list - True if annotation is staged CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_staged_flags RESTful: Method: GET URL: /api/annot/staged/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> gid_list = get_annot_staged_flags(ibs, aid_list) >>> result = str(gid_list) >>> print(result) """ try: annot_staged_flag_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_STAGED_FLAG,), aid_list ) except KeyError: # Support for old databases and migration annot_staged_flag_list = [False] * len(aid_list) annot_staged_flag_list = [ False if annot_staged_flag is None else bool(annot_staged_flag) for annot_staged_flag in annot_staged_flag_list ] return annot_staged_flag_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/staged/uuid/', methods=['GET']) def get_annot_staged_uuids(ibs, aid_list): r""" Returns: list: annot_uuid_list a list of image uuids by aid RESTful: Method: GET URL: /api/annot/staged/uuid/ """ annot_uuid_list = ibs.db.get(const.ANNOTATION_TABLE, ('annot_staged_uuid',), aid_list) return annot_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/staged/user/', methods=['GET']) def get_annot_staged_user_ids(ibs, aid_list): r""" returns if an annotation is staged Args: ibs (IBEISController): wbia controller object aid_list (int): list of annotation ids Returns: list: annot_staged_user_id_list - True if annotation is staged CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_staged_user_ids RESTful: Method: GET URL: /api/annot/staged/user/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> gid_list = get_annot_staged_user_ids(ibs, aid_list) >>> result = str(gid_list) >>> print(result) """ annot_staged_user_id_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_STAGED_USER_ID,), aid_list ) return annot_staged_user_id_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/staged/metadata/', methods=['GET']) def get_annot_staged_metadata(ibs, aid_list, return_raw=False): r""" Returns: list_ (list): annot metadata dictionary RESTful: Method: GET URL: /api/annot/staged/metadata/ """ metadata_str_list = ibs.db.get( const.ANNOTATION_TABLE, ('annot_staged_metadata_json',), aid_list ) metadata_list = [] for metadata_str in metadata_str_list: if metadata_str in [None, '']: metadata_dict = {} else: metadata_dict = metadata_str if return_raw else ut.from_json(metadata_str) metadata_list.append(metadata_dict) return metadata_list
[docs]@register_ibs_method def set_annot_viewpoint_int(ibs, aids, view_ints, _code_update=True): view_ints = list(view_ints) ibs.db.set(const.ANNOTATION_TABLE, ('annot_viewpoint_int',), view_ints, id_iter=aids) if _code_update: # oops didn't realize there was an old structure view_codes = ut.take(ibs.const.VIEW.INT_TO_CODE, view_ints) ibs.set_annot_viewpoints(aids, view_codes, _code_update=False)
[docs]@register_ibs_method def get_annot_viewpoint_code(ibs, aids): """ Doctest: >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[::3] >>> result = get_annot_viewpoint_code(ibs, aid_list) >>> print(result) ['left', 'left', 'unknown', 'left', 'unknown'] """ view_ints = ibs.get_annot_viewpoint_int(aids) return ut.dict_take(ibs.const.VIEW.INT_TO_CODE, view_ints)
def _ensure_viewpoint_to_code(view_codes): view_codes = [const.VIEW.CODE.UNKNOWN if v is None else v for v in view_codes] return view_codes def _ensure_viewpoint_to_int(view_codes): view_codes = [const.VIEW.CODE.UNKNOWN if v is None else v for v in view_codes] view_ints = ut.dict_take(const.VIEW.CODE_TO_INT, view_codes) return view_ints
[docs]@register_ibs_method def set_annot_viewpoint_code(ibs, aids, view_codes, _code_update=True): view_ints = _ensure_viewpoint_to_int(view_codes) ibs.set_annot_viewpoint_int(aids, view_ints, _code_update=_code_update)
[docs]@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator(const.ANNOTATION_TABLE, [ANNOT_YAW], rowidx=0) @register_api('/api/annot/yaw/', methods=['PUT']) def set_annot_yaws(ibs, aid_list, yaw_list, input_is_degrees=False): r""" Sets the yaw of a list of chips by aid DEPRICATE A yaw is the yaw of the annotation in radians yaw is inverted. Will be fixed soon. Note: The following views have these angles of yaw: left side - 0.00 tau radians front side - 0.25 tau radians right side - 0.50 tau radians back side - 0.75 tau radians (tau = 2 * pi) SeeAlso: wbia.const.VIEWTEXT_TO_YAW_RADIANS References: http://upload.wikimedia.org/wikipedia/commons/7/7e/Rollpitchyawplain.png RESTful: Method: PUT URL: /api/annot/yaw/ """ id_iter = ((aid,) for aid in aid_list) # yaw_list = [-1 if yaw is None else yaw for yaw in yaw_list] if input_is_degrees: yaw_list = [-1 if yaw is None else ut.deg_to_rad(yaw) for yaw in yaw_list] # assert all([0.0 <= yaw < 2 * np.pi or yaw == -1.0 for yaw in yaw_list]) val_iter = ((yaw,) for yaw in yaw_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_YAW,), val_iter, id_iter) ibs.update_annot_visual_uuids(aid_list) ibs.depc_annot.notify_root_changed(aid_list, 'yaws') # Also set the annotation's viewpoints viewpoint_list = ibs.get_annot_yaw_texts(aid_list) ibs.set_annot_viewpoints(aid_list, viewpoint_list, _yaw_update=False)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/viewpoint/', methods=['PUT']) def set_annot_viewpoints( ibs, aid_list, viewpoint_list, purge_cache=True, only_allow_known=True, _yaw_update=False, _code_update=True, ): r""" Sets the viewpoint of the annotation RESTful: Method: PUT URL: /api/annot/viewpoint/ """ viewpoint_list = list(viewpoint_list) if only_allow_known: isvalid = [v is None or v in const.VIEW.CODE_TO_INT for v in viewpoint_list] aid_list = ut.compress(aid_list, isvalid) viewpoint_list = ut.compress(viewpoint_list, isvalid) if purge_cache: current_viewpoint_list = ibs.get_annot_viewpoints(aid_list) val_iter = zip(viewpoint_list) id_iter = zip(aid_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_VIEWPOINT,), val_iter, id_iter) if purge_cache: flag_list = [ viewpoint != current_viewpoint for viewpoint, current_viewpoint in zip( viewpoint_list, current_viewpoint_list ) ] update_aid_list = ut.compress(aid_list, flag_list) try: ibs.wbia_plugin_curvrank_delete_cache_optimized( update_aid_list, 'CurvRankDorsal' ) except Exception: message = 'Could not purge CurvRankDorsal cache for viewpoint' # raise RuntimeError(message) logger.info(message) try: ibs.wbia_plugin_curvrank_v2_delete_cache_optimized( update_aid_list, 'CurvRankTwoDorsal' ) except Exception: message = 'Could not purge CurvRankTwoDorsal cache for viewpoint' # raise RuntimeError(message) logger.info(message) try: ibs.wbia_plugin_curvrank_delete_cache_optimized( update_aid_list, 'CurvRankFinfindrHybridDorsal' ) except Exception: message = 'Could not purge CurvRankFinfindrHybridDorsal cache for viewpoint' # raise RuntimeError(message) logger.info(message) # oops didn't realize there was a structure already here for this if _code_update: ibs.set_annot_viewpoint_code(aid_list, viewpoint_list, _code_update=False) if _yaw_update: # Set yaws, if in the correct plane, else None yaw_list = ut.dict_take(const.VIEWTEXT_TO_YAW_RADIANS, viewpoint_list, None) ibs.set_annot_yaws(aid_list, yaw_list)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/note/', methods=['GET']) def get_annot_notes(ibs, aid_list): r""" Returns: annotation_notes_list (list): a list of annotation notes RESTful: Method: GET URL: /api/annot/note/ """ annotation_notes_list = ibs.db.get(const.ANNOTATION_TABLE, (ANNOT_NOTE,), aid_list) return annotation_notes_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/metadata/', methods=['GET']) def get_annot_metadata(ibs, aid_list, return_raw=False): r""" Returns: list_ (list): annot metadata dictionary RESTful: Method: GET URL: /api/annot/metadata/ """ metadata_str_list = ibs.db.get( const.ANNOTATION_TABLE, ('annot_metadata_json',), aid_list ) metadata_list = [] for metadata_str in metadata_str_list: if metadata_str in [None, '']: metadata_dict = {} else: metadata_dict = metadata_str if return_raw else ut.from_json(metadata_str) metadata_list.append(metadata_dict) return metadata_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_num_groundtruth( ibs, aid_list, is_exemplar=None, noself=True, daid_list=None ): r""" Returns: list_ (list): number of other chips with the same name CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_num_groundtruth python -m wbia.control.manual_annot_funcs --test-get_annot_num_groundtruth:0 python -m wbia.control.manual_annot_funcs --test-get_annot_num_groundtruth:1 Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> noself = True >>> result = get_annot_num_groundtruth(ibs, aid_list, noself=noself) >>> print(result) [0, 1, 1, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0] Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> noself = False >>> result = get_annot_num_groundtruth(ibs, aid_list, noself=noself) >>> print(result) [1, 2, 2, 1, 2, 2, 1, 1, 1, 1, 1, 1, 1] """ # TODO: Optimize groundtruth_list = ibs.get_annot_groundtruth( aid_list, is_exemplar=is_exemplar, noself=noself, daid_list=daid_list ) nGt_list = list(map(len, groundtruth_list)) return nGt_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/num/vert/', methods=['GET']) def get_annot_num_verts(ibs, aid_list): r""" Returns: nVerts_list (list): the number of vertices that form the polygon of each chip RESTful: Method: GET URL: /api/annot/num/vert/ """ nVerts_list = ibs.db.get(const.ANNOTATION_TABLE, (ANNOT_NUM_VERTS,), aid_list) return nVerts_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/parent/aid/', methods=['GET']) def get_annot_parent_aid(ibs, aid_list): r""" Returns: list_ (list): a list of parent (in terms of parts) annotation rowids. """ annot_parent_rowid_list = ibs.db.get( const.ANNOTATION_TABLE, (ANNOT_PARENT_ROWID,), aid_list ) return annot_parent_rowid_list
[docs]@register_ibs_method @accessor_decors.getter_1toM @register_api('/api/annot/part/rowid/', methods=['GET']) def get_annot_part_rowids(ibs, aid_list, is_staged=False): r""" Returns: list_ (list): a list of part rowids for each image by aid Args: ibs (IBEISController): wbia controller object aid_list (list): Returns: list: part_rowids_list RESTful: Method: GET URL: /api/annot/part/rowid/ """ # FIXME: This index should when the database is defined. # Ensure that an index exists on the image column of the annotation table with ibs.db.connect() as conn: conn.execute( """ CREATE INDEX IF NOT EXISTS aid_to_part_rowids ON parts (annot_rowid); """ ) # The index maxes the following query very efficient part_rowids_list = ibs.db.get( ibs.const.PART_TABLE, (PART_ROWID,), aid_list, id_colname=ANNOT_ROWID, unpack_scalars=False, ) part_rowids_list = [ ibs.filter_part_set(part_rowid_list, is_staged=is_staged) for part_rowid_list in part_rowids_list ] return part_rowids_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 @accessor_decors.cache_getter( const.ANNOTATION_TABLE, NAME_ROWID, cfgkeys=['distinguish_unknowns'] ) # @register_api('/api/annot/name/rowid/', methods=['GET']) def get_annot_name_rowids(ibs, aid_list, distinguish_unknowns=True, assume_unique=False): r""" Returns: list_ (list): the name id of each annotation. CommandLine: python -m wbia.control.manual_annot_funcs --exec-get_annot_name_rowids Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> from wbia import constants as const >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> distinguish_unknowns = True >>> nid_arr1 = np.array(ibs.get_annot_name_rowids(aid_list, distinguish_unknowns=distinguish_unknowns)) >>> nid_arr2 = np.array(ibs.get_annot_name_rowids(aid_list, distinguish_unknowns=False)) >>> nid_arr2 = np.array(ibs.get_annot_name_rowids(None, distinguish_unknowns=True)) >>> assert const.UNKNOWN_LBLANNOT_ROWID == 0 >>> assert np.all(nid_arr1[np.where(const.UNKNOWN_LBLANNOT_ROWID == nid_arr2)[0]] < 0) """ id_iter = aid_list colnames = (NAME_ROWID,) nid_list_ = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid', assume_unique=assume_unique, ) if distinguish_unknowns: nid_list = [ (None if aid is None else -aid) if nid == const.UNKNOWN_LBLANNOT_ROWID or nid is None else nid for nid, aid in zip(nid_list_, aid_list) ] else: nid_list = [ const.UNKNOWN_LBLANNOT_ROWID if nid is None else nid for nid in nid_list_ ] return nid_list
[docs]@register_ibs_method @register_api('/api/annot/name/rowid/', methods=['GET']) def get_annot_nids(ibs, aid_list, distinguish_unknowns=True): r""" alias RESTful: Method: GET URL: /api/annot/name/rowid/ """ return ibs.get_annot_name_rowids(aid_list, distinguish_unknowns=distinguish_unknowns)
[docs]@register_ibs_method @register_api('/api/annot/name/uuid/', methods=['GET']) def get_annot_name_uuids(ibs, aid_list, **kwargs): r""" alias RESTful: Method: GET URL: /api/annot/name/uuid/ """ nid_list = ibs.get_annot_name_rowids(aid_list, **kwargs) name_uuid_list = ibs.get_name_uuids(nid_list) return name_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/name/', methods=['GET']) def get_annot_names(ibs, aid_list, distinguish_unknowns=False): r""" alias """ return ibs.get_annot_name_texts(aid_list, distinguish_unknowns=distinguish_unknowns)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/name/text/', methods=['GET']) def get_annot_name_texts(ibs, aid_list, distinguish_unknowns=False): r""" Args: aid_list (list): Returns: list or strs: name_list. e.g: ['fred', 'sue', ...] for each annotation identifying the individual RESTful: Method: GET URL: /api/annot/name/text/ CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_name_texts Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[::2] >>> result = ut.repr2(get_annot_name_texts(ibs, aid_list), nl=False) >>> print(result) ['____', 'easy', 'hard', 'jeff', '____', '____', 'zebra'] Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[::2] >>> result = ut.repr2(get_annot_name_texts(ibs, aid_list, True), nl=False) >>> print(result) ['____1', 'easy', 'hard', 'jeff', '____9', '____11', 'zebra'] """ aid_list = list(aid_list) nid_list = ibs.get_annot_name_rowids(aid_list) name_list = ibs.get_name_texts(nid_list) if distinguish_unknowns: name_list = [ name if nid >= 0 else name + '%d' % -nid for nid, name in zip(nid_list, name_list) ] # name_list = ibs.get_annot_lblannot_value_of_lbltype(aid_list, # const.INDIVIDUAL_KEY, ibs.get_name_texts) return name_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/species/', methods=['GET'], __api_plural_check__=False) def get_annot_species(ibs, aid_list): r""" alias RESTful: Method: GET URL: /api/annot/species/ """ return ibs.get_annot_species_texts(aid_list)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/species/text/', methods=['GET'], __api_plural_check__=False) def get_annot_species_texts(ibs, aid_list): r""" Args: aid_list (list): Returns: list : species_list - a list of strings ['plains_zebra', 'grevys_zebra', ...] for each annotation identifying the species CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_species_texts Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[1::3] >>> result = ut.repr2(get_annot_species_texts(ibs, aid_list), nl=False) >>> print(result) ['zebra_plains', 'zebra_plains', '____', 'bear_polar'] Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('PZ_MTEST') >>> aid_list = ibs.get_valid_aids() >>> species_list = get_annot_species_texts(ibs, aid_list) >>> result = ut.repr2(list(set(species_list)), nl=False) >>> print(result) ['zebra_plains'] RESTful: Method: GET URL: /api/annot/species/text/ """ species_rowid_list = ibs.get_annot_species_rowids(aid_list) speceis_text_list = ibs.get_species_texts(species_rowid_list) # speceis_text_list = ibs.get_annot_lblannot_value_of_lbltype( # aid_list, const.SPECIES_KEY, ibs.get_species) return speceis_text_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 @accessor_decors.cache_getter(const.ANNOTATION_TABLE, SPECIES_ROWID) @register_api('/api/annot/species/rowid/', methods=['GET'], __api_plural_check__=False) def get_annot_species_rowids(ibs, aid_list): r""" species_rowid_list <- annot.species_rowid[aid_list] gets data from the "native" column "species_rowid" in the "annot" table Args: aid_list (list): Returns: list: species_rowid_list RESTful: Method: GET URL: /api/annot/species/rowid/ """ id_iter = aid_list colnames = (SPECIES_ROWID,) species_rowid_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid' ) return species_rowid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/species/uuid/', methods=['GET'], __api_plural_check__=False) def get_annot_species_uuids(ibs, aid_list): r""" species_rowid_list <- annot.species_rowid[aid_list] Args: aid_list (list): Returns: list: species_uuid_list RESTful: Method: GET URL: /api/annot/species/uuid/ """ species_rowid_list = ibs.get_annot_species_rowids(aid_list) species_uuid_list = ibs.get_species_uuids(species_rowid_list) return species_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/image/name/', methods=['GET']) def get_annot_image_names(ibs, aid_list): r""" Args: aid_list (list): Returns: list of strs: gname_list the image names of each annotation RESTful: Method: GET URL: /api/annot/image/name/ """ gid_list = ibs.get_annot_gids(aid_list) gname_list = ibs.get_image_gnames(gid_list) return gname_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/image/unixtime/', methods=['GET']) def get_annot_image_unixtimes(ibs, aid_list, **kwargs): r""" Args: aid_list (list): Returns: list: unixtime_list RESTful: Method: GET URL: /api/annot/image/unixtime/ """ gid_list = ibs.get_annot_gids(aid_list) unixtime_list = ibs.get_image_unixtime(gid_list, **kwargs) return unixtime_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 # @register_api('/api/annot/image/unixtime/float/', methods=['GET']) def get_annot_image_unixtimes_asfloat(ibs, aid_list): r""" Args: ibs (IBEISController): wbia controller object aid_list (list): list of annotation rowids Returns: list: unixtime_list CommandLine: python -m wbia.control.manual_annot_funcs --exec-get_annot_image_unixtimes_asfloat --show --db PZ_MTEST Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ibs.get_valid_aids() >>> unixtime_list = get_annot_image_unixtimes_asfloat(ibs, aid_list) >>> result = ('unixtime_list = %s' % (str(unixtime_list),)) >>> print(result) >>> ut.quit_if_noshow() >>> import wbia.plottool as pt >>> ut.show_if_requested() """ unixtime_list = np.array(ibs.get_annot_image_unixtimes(aid_list), dtype=np.float64) unixtime_list[unixtime_list == -1] = np.nan return unixtime_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_image_datetime_str(ibs, aid_list): r""" Args: ibs (IBEISController): wbia controller object aid_list (int): list of annotation ids Returns: list: datetime_list CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_image_datetime_str Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> datetime_list = get_annot_image_datetime_str(ibs, aid_list) >>> result = str(datetime_list) >>> print(result) """ gid_list = ibs.get_annot_gids(aid_list) datetime_list = ibs.get_image_datetime_str(gid_list) return datetime_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 @register_api('/api/annot/image/gps/', methods=['GET'], __api_plural_check__=False) def get_annot_image_gps(ibs, aid_list): r""" Args: aid_list (list): Returns: list: unixtime_list RESTful: Method: GET URL: /api/annot/image/gps/ """ gid_list = ibs.get_annot_gids(aid_list) gps_list = ibs.get_image_gps(gid_list) return gps_list
[docs]@register_ibs_method @ut.accepts_numpy @accessor_decors.getter_1to1 @register_api('/api/annot/image/gps2/', methods=['GET'], __api_plural_check__=False) def get_annot_image_gps2(ibs, aid_list): r""" fixes the (-1, -1) issue. returns nan instead. """ gid_list = ibs.get_annot_gids(aid_list) gps_list = ibs.get_image_gps2(gid_list) return gps_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/image/file/path/', methods=['GET']) def get_annot_image_paths(ibs, aid_list): r""" Args: aid_list (list): Returns: list of strs: gpath_list the image paths of each annotation RESTful: Method: GET URL: /api/annot/image/file/path/ """ gid_list = ibs.get_annot_gids(aid_list) try: ut.assert_all_not_None(gid_list, 'gid_list') except AssertionError: logger.info('[!get_annot_image_paths] aids=' + ut.repr4(aid_list)) logger.info('[!get_annot_image_paths] gids=' + ut.repr4(gid_list)) raise gpath_list = ibs.get_image_paths(gid_list) ut.assert_all_not_None(gpath_list, 'gpath_list') return gpath_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/image/uuid/', methods=['GET']) def get_annot_image_uuids(ibs, aid_list): r""" Args: aid_list (list): Returns: list: image_uuid_list CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_image_uuids --enableall RESTful: Method: GET URL: /api/annot/image/uuid/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:1] >>> result = get_annot_image_uuids(ibs, aid_list) >>> print(result) [UUID('66ec193a-1619-b3b6-216d-1784b4833b61')] """ gid_list = ibs.get_annot_gids(aid_list) ut.assert_all_not_None(gid_list, 'gid_list') image_uuid_list = ibs.get_image_uuids(gid_list) return image_uuid_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/images/', methods=['GET']) def get_annot_images(ibs, aid_list): r""" Args: aid_list (list): Returns: list of ndarrays: the images of each annotation CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annot_images Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:1] >>> image_list = ibs.get_annot_images(aid_list) >>> result = str(list(map(np.shape, image_list))) >>> print(result) [(715, 1047, 3)] """ gid_list = ibs.get_annot_gids(aid_list) image_list = ibs.get_images(gid_list) return image_list
#### SETTERS #### # NOQA
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/bbox/', methods=['PUT']) def set_annot_bboxes(ibs, aid_list, bbox_list, delete_thumbs=True, **kwargs): r""" Sets bboxes of a list of annotations by aid, Args: aid_list (list of rowids): list of annotation rowids bbox_list (list of (x, y, w, h)): new bounding boxes for each aid Note: set_annot_bboxes is a proxy for set_annot_verts RESTful: Method: PUT URL: /api/annot/bbox/ """ from vtool import geometry # changing the bboxes also changes the bounding polygon vert_list = geometry.verts_list_from_bboxes_list(bbox_list) # naively overwrite the bounding polygon with a rectangle - for now trust the user! ibs.set_annot_verts(aid_list, vert_list, delete_thumbs=delete_thumbs, **kwargs)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/detect/confidence/', methods=['PUT']) def set_annot_detect_confidence(ibs, aid_list, confidence_list): r""" Sets annotation notes RESTful: Method: PUT URL: /api/annot/detect/confidence/ """ id_iter = ((aid,) for aid in aid_list) val_iter = ((confidence,) for confidence in confidence_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_detect_confidence',), val_iter, id_iter)
[docs]@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator( const.ANNOTATION_TABLE, [ANNOT_EXEMPLAR_FLAG], rowidx=0 ) @accessor_decors.cache_invalidator( const.IMAGESET_TABLE, ['percent_names_with_exemplar_str'] ) @register_api('/api/annot/exemplar/', methods=['PUT']) def set_annot_exemplar_flags(ibs, aid_list, flag_list): r""" Sets if an annotation is an exemplar RESTful: Method: PUT URL: /api/annot/exemplar/ """ id_iter = ((aid,) for aid in aid_list) val_iter = ((flag,) for flag in flag_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_EXEMPLAR_FLAG,), val_iter, id_iter)
[docs]@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator(const.ANNOTATION_TABLE, [NAME_ROWID], rowidx=0) @accessor_decors.cache_invalidator( const.IMAGESET_TABLE, ['percent_names_with_exemplar_str'] ) # @register_api('/api/annot/name/rowid/', methods=['PUT']) def set_annot_name_rowids( ibs, aid_list, name_rowid_list, notify_wildbook=True, assert_wildbook=False ): r""" name_rowid_list -> annot.name_rowid[aid_list] Sets names/nids of a list of annotations. Args: aid_list (list): name_rowid_list (list): Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:2] >>> # check clean state >>> ut.assert_eq(ibs.get_annot_names(aid_list), ['____', 'easy']) >>> ut.assert_eq(ibs.get_annot_exemplar_flags(aid_list), [0, 1]) >>> # run function >>> name_list = ['easy', '____'] >>> name_rowid_list = ibs.get_name_rowids_from_text(name_list) >>> ibs.set_annot_name_rowids(aid_list, name_rowid_list) >>> # check results >>> ut.assert_eq(ibs.get_annot_names(aid_list), ['easy', '____']) >>> ut.assert_eq(ibs.get_annot_exemplar_flags(aid_list), [0, 0]) >>> # restore database state >>> ibs.set_annot_names(aid_list, ['____', 'easy']) >>> ibs.set_annot_exemplar_flags(aid_list, [0, 1]) >>> ut.assert_eq(ibs.get_annot_names(aid_list), ['____', 'easy']) >>> ut.assert_eq(ibs.get_annot_exemplar_flags(aid_list), [0, 1]) """ assert len(aid_list) == len(name_rowid_list), 'misaligned' import wbia if notify_wildbook and wbia.ENABLE_WILDBOOK_SIGNAL: try: ibs.wildbook_signal_annot_name_changes(aid_list) except requests.exceptions.ConnectionError: if assert_wildbook: raise IOError('Cannot connect to WB for name updates') # ibsfuncs.assert_lblannot_rowids_are_type(ibs, name_rowid_list, # ibs.lbltype_ids[const.INDIVIDUAL_KEY]) id_iter = aid_list colnames = (NAME_ROWID,) # WE NEED TO PERFORM A SPECIAL CHECK. ANY ANIMAL WHICH IS GIVEN AN UNKONWN # NAME MUST HAVE ITS EXEMPLAR FLAG SET TO FALSE will_be_unknown_flag_list = [ nid == const.UNKNOWN_NAME_ROWID for nid in name_rowid_list ] if any(will_be_unknown_flag_list): # remove exemplar status from any annotations that will become unknown will_be_unknown_aids = ut.compress(aid_list, will_be_unknown_flag_list) ibs.set_annot_exemplar_flags( will_be_unknown_aids, [False] * len(will_be_unknown_aids) ) ibs.db.set(const.ANNOTATION_TABLE, colnames, name_rowid_list, id_iter) # postset nids ibs.update_annot_semantic_uuids(aid_list) # TODO: flag name rowid update # TODO: flag when the actual name changes as well? ibs.depc_annot.notify_root_changed(aid_list, 'name')
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/name/', methods=['PUT']) def set_annot_names(ibs, aid_list, name_list, **kwargs): r""" Sets the attrlbl_value of type(INDIVIDUAL_KEY) Sets names/nids of a list of annotations. CommandLine: python -m wbia.control.manual_annot_funcs --test-set_annot_names --enableall RESTful: Method: PUT URL: /api/annot/name/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids() >>> name_list1 = get_annot_names(ibs, aid_list) >>> name_list2 = [name + '_TESTAUG' for name in name_list1] >>> set_annot_names(ibs, aid_list, name_list2) >>> name_list3 = get_annot_names(ibs, aid_list) >>> set_annot_names(ibs, aid_list, name_list1) >>> name_list4 = get_annot_names(ibs, aid_list) >>> assert name_list2 == name_list3 >>> assert name_list4 == name_list1 >>> assert name_list4 != name_list2 >>> print(result) """ # ibs.get_nids_from_text assert len(aid_list) == len(name_list) assert not isinstance(name_list, str) # name_rowid_list = ibs.get_name_rowids_from_text(name_list, ensure=True) assert not any( [name == '' for name in name_list] ), 'cannot change name to empty string use ____ for unknown.' name_rowid_list = ibs.add_names(name_list) ibs.set_annot_name_rowids(aid_list, name_rowid_list, **kwargs)
[docs]@register_ibs_method @accessor_decors.getter_1to1 def set_annot_name_texts(ibs, aid_list, name_list): r""" alias RESTful: Method: GET URL: /api/annot/name/ """ return ibs.set_annot_names(aid_list, name_list)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/species/', methods=['PUT'], __api_plural_check__=False) def set_annot_species(ibs, aid_list, species_text_list, **kwargs): r""" Sets species/speciesids of a list of annotations. Convenience function for set_annot_lblannot_from_value RESTful: Method: PUT URL: /api/annot/species/ """ # ibs.get_nids_from_text species_rowid_list = ibs.get_species_rowids_from_text(species_text_list, **kwargs) ibs.set_annot_species_rowids(aid_list, species_rowid_list)
[docs]@register_ibs_method @accessor_decors.setter def set_annot_species_and_notify(ibs, *args, **kwargs): # for gui ibs.set_annot_species(*args, **kwargs) ibs.notify_observers()
[docs]@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator(const.ANNOTATION_TABLE, [SPECIES_ROWID], rowidx=0) @register_api('/api/annot/species/rowid/', methods=['PUT'], __api_plural_check__=False) def set_annot_species_rowids(ibs, aid_list, species_rowid_list): r""" species_rowid_list -> annot.species_rowid[aid_list] Sets species/speciesids of a list of annotations. Args: aid_list species_rowid_list RESTful: Method: PUT URL: /api/annot/species/rowid/ """ id_iter = aid_list colnames = (SPECIES_ROWID,) ibs.db.set(const.ANNOTATION_TABLE, colnames, species_rowid_list, id_iter) ibs.update_annot_semantic_uuids(aid_list) ibs.depc_annot.notify_root_changed(aid_list, 'species')
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/note/', methods=['PUT']) def set_annot_notes(ibs, aid_list, notes_list): r""" Sets annotation notes RESTful: Method: PUT URL: /api/annot/note/ """ id_iter = ((aid,) for aid in aid_list) val_iter = ((notes,) for notes in notes_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_NOTE,), val_iter, id_iter)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/metadata/', methods=['PUT']) def set_annot_metadata(ibs, aid_list, metadata_dict_list): r""" Sets the annot's metadata using a metadata dictionary RESTful: Method: PUT URL: /api/annot/metadata/ CommandLine: python -m wbia.control.manual_annot_funcs --test-set_annot_metadata Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> import random >>> # build test data >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:1] >>> metadata_dict_list = [ >>> {'test': random.uniform(0.0, 1.0)}, >>> ] >>> print(ut.repr2(metadata_dict_list)) >>> ibs.set_annot_metadata(aid_list, metadata_dict_list) >>> # verify results >>> metadata_dict_list_ = ibs.get_annot_metadata(aid_list) >>> print(ut.repr2(metadata_dict_list_)) >>> assert metadata_dict_list == metadata_dict_list_ >>> metadata_str_list = [ut.to_json(metadata_dict) for metadata_dict in metadata_dict_list] >>> print(ut.repr2(metadata_str_list)) >>> metadata_str_list_ = ibs.get_annot_metadata(aid_list, return_raw=True) >>> print(ut.repr2(metadata_str_list_)) >>> assert metadata_str_list == metadata_str_list_ """ id_iter = ((aid,) for aid in aid_list) metadata_str_list = [] for metadata_dict in metadata_dict_list: metadata_str = ut.to_json(metadata_dict) metadata_str_list.append(metadata_str) val_list = ((metadata_str,) for metadata_str in metadata_str_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_metadata_json',), val_list, id_iter)
[docs]@register_ibs_method @accessor_decors.setter # @register_api('/api/annot/parent/rowid/', methods=['PUT']) def set_annot_parent_rowid(ibs, aid_list, parent_aid_list): r""" Sets the annotation's parent aid. TODO DEPRICATE IN FAVOR OF SEPARATE PARTS TABLE RESTful: Method: PUT URL: /api/annot/parent/rowid/ """ id_iter = ((aid,) for aid in aid_list) val_iter = ((parent_aid,) for parent_aid in parent_aid_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_PARENT_ROWID,), val_iter, id_iter)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/theta/', methods=['PUT']) def set_annot_thetas( ibs, aid_list, theta_list, delete_thumbs=True, update_visual_uuids=True, notify_root=True, ): r""" Sets thetas of a list of chips by aid RESTful: Method: PUT URL: /api/annot/theta/ """ id_iter = ((aid,) for aid in aid_list) val_list = ((theta,) for theta in theta_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_THETA,), val_list, id_iter) if delete_thumbs: ibs.delete_annot_chips(aid_list) # Changing theta redefines the chips ibs.delete_annot_imgthumbs(aid_list) if update_visual_uuids: ibs.update_annot_visual_uuids(aid_list) if notify_root: ibs.depc_annot.notify_root_changed(aid_list, 'theta', force_delete=True)
def _update_annot_rotate_fix_bbox(bbox): (xtl, ytl, w, h) = bbox diffx = int(round((w / 2.0) - (h / 2.0))) diffy = int(round((h / 2.0) - (w / 2.0))) xtl, ytl, w, h = xtl + diffx, ytl + diffy, h, w bbox = (xtl, ytl, w, h) return bbox
[docs]def update_annot_rotate_90(ibs, aid_list, direction): from wbia.constants import PI, TAU if isinstance(direction, str): direction = direction.lower() if direction in ['left', 'l', -1]: val = 1.0 elif direction in ['right', 'r', 1]: val = -1.0 else: raise ValueError('Invalid direction supplied') theta_list = ibs.get_annot_thetas(aid_list) theta_list = [(theta + (val * PI / 2)) % TAU for theta in theta_list] ibs.set_annot_thetas(aid_list, theta_list) bbox_list = ibs.get_annot_bboxes(aid_list) bbox_list = [_update_annot_rotate_fix_bbox(bbox) for bbox in bbox_list] ibs.set_annot_bboxes(aid_list, bbox_list)
[docs]@register_ibs_method @register_api('/api/annot/rotate/left/', methods=['POST']) def update_annot_rotate_left_90(ibs, aid_list): return update_annot_rotate_90(ibs, aid_list, 'left')
[docs]@register_ibs_method @register_api('/api/annot/rotate/right/', methods=['POST']) def update_annot_rotate_right_90(ibs, aid_list): return update_annot_rotate_90(ibs, aid_list, 'right')
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/vert/', methods=['PUT']) def set_annot_verts( ibs, aid_list, verts_list, theta_list=None, interest_list=None, canonical_list=None, delete_thumbs=True, update_visual_uuids=True, notify_root=True, ): r""" Sets the vertices [(x, y), ...] of a list of chips by aid RESTful: Method: PUT URL: /api/annot/vert/ """ with ut.Timer('set_annot_verts'): with ut.Timer('set_annot_verts...config'): from vtool import geometry nInput = len(aid_list) # Compute data to set if isinstance(verts_list, np.ndarray): verts_list = verts_list.tolist() for index, vert_list in enumerate(verts_list): if isinstance(vert_list, np.ndarray): verts_list[index] = vert_list.tolist() num_verts_list = list(map(len, verts_list)) verts_as_strings = list(map(str, verts_list)) id_iter1 = ((aid,) for aid in aid_list) # also need to set the internal number of vertices val_iter1 = ( (num_verts, verts) for (num_verts, verts) in zip(num_verts_list, verts_as_strings) ) colnames = ( ANNOT_NUM_VERTS, ANNOT_VERTS, ) # SET VERTS in ANNOTATION_TABLE ibs.db.set( const.ANNOTATION_TABLE, colnames, val_iter1, id_iter1, nInput=nInput ) # changing the vertices also changes the bounding boxes bbox_list = geometry.bboxes_from_vert_list(verts_list) # new bboxes xtl_list, ytl_list, width_list, height_list = list(zip(*bbox_list)) with ut.Timer('set_annot_verts...bbox'): val_iter2 = zip(xtl_list, ytl_list, width_list, height_list) id_iter2 = ((aid,) for aid in aid_list) colnames = ( 'annot_xtl', 'annot_ytl', 'annot_width', 'annot_height', ) # SET BBOX in ANNOTATION_TABLE ibs.db.set( const.ANNOTATION_TABLE, colnames, val_iter2, id_iter2, nInput=nInput ) with ut.Timer('set_annot_verts...theta'): if theta_list: ibs.set_annot_thetas( aid_list, theta_list, delete_thumbs=False, update_visual_uuids=False, notify_root=False, ) with ut.Timer('set_annot_verts...interest'): if interest_list: ibs.set_annot_interest(aid_list, interest_list, delete_thumbs=False) with ut.Timer('set_annot_verts...canonical'): if canonical_list: ibs.set_annot_canonical(aid_list, canonical_list, delete_thumbs=False) with ut.Timer('set_annot_verts...thumbs'): if delete_thumbs: ibs.delete_annot_chips(aid_list) # INVALIDATE THUMBNAILS ibs.delete_annot_imgthumbs(aid_list) gid_list = list(set(ibs.get_annot_gids(aid_list))) config2_ = {'thumbsize': 221} ibs.delete_image_thumbs(gid_list, quiet=True, **config2_) with ut.Timer('set_annot_verts...visual uuids'): if update_visual_uuids: ibs.update_annot_visual_uuids(aid_list) with ut.Timer('set_annot_verts...roots'): if notify_root: ibs.depc_annot.notify_root_changed(aid_list, 'verts', force_delete=True) if theta_list: ibs.depc_annot.notify_root_changed( aid_list, 'theta', force_delete=True )
# PROBCHIP
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/probchip/fpath/', methods=['GET']) def get_annot_probchip_fpath(ibs, aid_list, config2_=None): r""" Returns paths to probability images. Args: ibs (IBEISController): wbia controller object aid_list (list): list of annotation rowids config2_ (dict): (default = None) Returns: list: probchip_fpath_list CommandLine: python -m wbia.control.manual_annot_funcs --exec-get_annot_probchip_fpath --show Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='PZ_MTEST') >>> aid_list = ibs.get_valid_aids()[0:10] >>> config2_ = {'fw_detector': 'cnn'} >>> probchip_fpath_list = get_annot_probchip_fpath(ibs, aid_list, config2_) >>> result = ('probchip_fpath_list = %s' % (str(probchip_fpath_list),)) >>> print(result) >>> ut.quit_if_noshow() >>> import wbia.plottool as pt >>> iteract_obj = pt.interact_multi_image.MultiImageInteraction(probchip_fpath_list, nPerPage=4) >>> iteract_obj.start() >>> ut.show_if_requested() """ probchip_fpath_list = ibs.depc_annot.get( 'probchip', aid_list, 'img', config=config2_, read_extern=False ) return probchip_fpath_list
# --- # NEW # ---
[docs]@register_ibs_method @accessor_decors.getter_1to1 @accessor_decors.cache_getter(const.ANNOTATION_TABLE, ANNOT_QUALITY) @register_api('/api/annot/quality/', methods=['GET']) def get_annot_qualities(ibs, aid_list, eager=True): r""" annot_quality_list <- annot.annot_quality[aid_list] gets data from the "native" column "annot_quality" in the "annot" table Args: aid_list (list): Returns: list: annot_quality_list TemplateInfo: Tgetter_table_column col = annot_quality tbl = annot SeeAlso: wbia.const.QUALITY_INT_TO_TEXT RESTful: Method: GET URL: /api/annot/quality/ Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> ibs, qreq_ = testdata_ibs() >>> aid_list = ibs._get_all_aids() >>> eager = True >>> annot_quality_list = ibs.get_annot_qualities(aid_list, eager=eager) >>> print('annot_quality_list = %r' % (annot_quality_list,)) >>> assert len(aid_list) == len(annot_quality_list) """ id_iter = aid_list colnames = (ANNOT_QUALITY,) annot_quality_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid', eager=eager ) return annot_quality_list
[docs]@register_ibs_method def get_annot_quality_int(ibs, aid_list, eager=True): """new alias""" return ibs.get_annot_qualities(aid_list, eager=eager)
[docs]@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator(const.ANNOTATION_TABLE, [ANNOT_QUALITY], rowidx=0) @register_api('/api/annot/quality/', methods=['PUT']) def set_annot_qualities(ibs, aid_list, annot_quality_list): r""" annot_quality_list -> annot.annot_quality[aid_list] A quality is an integer representing the following types: Args: aid_list annot_quality_list SeeAlso: wbia.const.QUALITY_INT_TO_TEXT RESTful: Method: PUT URL: /api/annot/quality/ """ id_iter = aid_list colnames = (ANNOT_QUALITY,) ibs.db.set(const.ANNOTATION_TABLE, colnames, annot_quality_list, id_iter)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/quality/text/', methods=['GET']) def get_annot_quality_texts(ibs, aid_list): r""" Auto-docstr for 'get_annot_quality_texts' RESTful: Method: GET URL: /api/annot/quality/text/ """ quality_list = ibs.get_annot_qualities(aid_list) quality_text_list = ut.dict_take(const.QUALITY_INT_TO_TEXT, quality_list) return quality_text_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 # @register_api('/api/annot/isjunk/', methods=['GET']) def get_annot_isjunk(ibs, aid_list): r""" Auto-docstr for 'get_annot_isjunk' """ qual_list = ibs.get_annot_qualities(aid_list) # isjunk_list = [qual == const.QUALITY_TEXT_TO_INT['junk'] for qual in qual_list] isjunk_list = [qual in const.QUALITY_TEXT_TO_INTS['junk'] for qual in qual_list] return isjunk_list
[docs]@register_ibs_method @register_api('/api/annot/quality/text/', methods=['PUT']) def set_annot_quality_texts(ibs, aid_list, quality_text_list): r""" Auto-docstr for 'set_annot_quality_texts' RESTful: Method: PUT URL: /api/annot/quality/text/ """ if not ut.isiterable(aid_list): aid_list = [aid_list] if isinstance(quality_text_list, str): quality_text_list = [quality_text_list] quality_list = ut.dict_take(const.QUALITY_TEXT_TO_INT, quality_text_list) ibs.set_annot_qualities(aid_list, quality_list)
[docs]@register_ibs_method @register_api('/api/annot/yaw/text/', methods=['PUT']) def set_annot_yaw_texts(ibs, aid_list, yaw_text_list): r""" Auto-docstr for 'set_annot_yaw_texts' DEPRICATE RESTful: Method: PUT URL: /api/annot/yaw/text/ """ if not ut.isiterable(aid_list): aid_list = [aid_list] if isinstance(yaw_text_list, str): yaw_text_list = [yaw_text_list] ibs.set_annot_viewpoint_code(aid_list, yaw_text_list)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/sex/', methods=['GET']) def get_annot_sex(ibs, aid_list, eager=True, nInput=None): r""" Auto-docstr for 'get_annot_sex' RESTful: Method: GET URL: /api/annot/sex/ """ nid_list = ibs.get_annot_nids(aid_list) sex_list = ibs.get_name_sex(nid_list) return sex_list
[docs]@register_ibs_method @register_api('/api/annot/sex/text/', methods=['GET']) def get_annot_sex_texts(ibs, aid_list, eager=True, nInput=None): r""" Auto-docstr for 'get_annot_sex_texts' RESTful: Method: GET URL: /api/annot/sex/text/ """ nid_list = ibs.get_annot_nids(aid_list) sex_text_list = ibs.get_name_sex_text(nid_list) return sex_text_list
[docs]@register_ibs_method @register_api('/api/annot/sex/', methods=['PUT']) def set_annot_sex(ibs, aid_list, name_sex_list, eager=True, nInput=None): r""" Auto-docstr for 'set_annot_sex' RESTful: Method: PUT URL: /api/annot/sex/ """ nid_list = ibs.get_annot_nids(aid_list) flag_list = [nid is not None for nid in nid_list] nid_list = ut.compress(nid_list, flag_list) name_sex_list = ut.compress(name_sex_list, flag_list) ibs.set_name_sex(nid_list, name_sex_list)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/sex/text/', methods=['PUT']) def set_annot_sex_texts(ibs, aid_list, name_sex_text_list, eager=True, nInput=None): r""" Auto-docstr for 'set_annot_sex_texts' RESTful: Method: PUT URL: /api/annot/sex/text/ """ nid_list = ibs.get_annot_nids(aid_list) flag_list = [nid is not None for nid in nid_list] nid_list = ut.compress(nid_list, flag_list) name_sex_text_list = ut.compress(name_sex_text_list, flag_list) ibs.set_name_sex_text(nid_list, name_sex_text_list)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/age/months/min/', methods=['GET'], __api_plural_check__=False) def get_annot_age_months_est_min(ibs, aid_list, eager=True, nInput=None): r""" annot_age_months_est_min_list <- annot.annot_age_months_est_min[aid_list] gets data from the "native" column "annot_age_months_est_min" in the "annot" table Args: aid_list (list): Returns: list: annot_age_months_est_min_list RESTful: Method: GET URL: /api/annot/age/months/min/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> ibs, config2_ = testdata_ibs() >>> aid_list = ibs._get_all_aids() >>> eager = True >>> annot_age_months_est_min_list = ibs.get_annot_age_months_est_min(aid_list, eager=eager) >>> assert len(aid_list) == len(annot_age_months_est_min_list) """ id_iter = aid_list colnames = (ANNOT_AGE_MONTHS_EST_MIN,) annot_age_months_est_min_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid', eager=eager, nInput=nInput, ) return annot_age_months_est_min_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/age/months/max/', methods=['GET'], __api_plural_check__=False) def get_annot_age_months_est_max(ibs, aid_list, eager=True, nInput=None): r""" annot_age_months_est_max_list <- annot.annot_age_months_est_max[aid_list] gets data from the "native" column "annot_age_months_est_max" in the "annot" table Args: aid_list (list): Returns: list: annot_age_months_est_max_list RESTful: Method: GET URL: /api/annot/age/months/max/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> ibs, config2_ = testdata_ibs() >>> aid_list = ibs._get_all_aids() >>> eager = True >>> annot_age_months_est_max_list = ibs.get_annot_age_months_est_max(aid_list, eager=eager) >>> assert len(aid_list) == len(annot_age_months_est_max_list) """ id_iter = aid_list colnames = (ANNOT_AGE_MONTHS_EST_MAX,) annot_age_months_est_max_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid', eager=eager, nInput=nInput, ) return annot_age_months_est_max_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/age/months/', methods=['GET'], __api_plural_check__=False) def get_annot_age_months_est(ibs, aid_list, eager=True, nInput=None): r""" annot_age_months_est_list <- annot.annot_age_months_est[aid_list] gets data from the annotation's native age in months Args: aid_list (list): Returns: list: annot_age_months_est_list RESTful: Method: GET URL: /api/annot/age/months/ """ annot_age_months_est_min_list = ibs.get_annot_age_months_est_min(aid_list) annot_age_months_est_max_list = ibs.get_annot_age_months_est_max(aid_list) annot_age_months_est_list = list( zip(annot_age_months_est_min_list, annot_age_months_est_max_list) ) return annot_age_months_est_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api( '/api/annot/age/months/min/text/', methods=['GET'], __api_plural_check__=False ) def get_annot_age_months_est_min_texts(ibs, aid_list, eager=True, nInput=None): r""" annot_age_months_est_min_texts_list <- annot.annot_age_months_est_min_texts[aid_list] gets string versions of the annotation's native min age in months Args: aid_list (list): Returns: list: annot_age_months_est_min_list RESTful: Method: GET URL: /api/annot/age/months/min/text/ """ annot_age_months_est_min_list = ibs.get_annot_age_months_est_min(aid_list) annot_age_months_est_min_text_list = [ 'Unknown' if age_min in [None, -1] else '%d Months' % (age_min,) for age_min in annot_age_months_est_min_list ] return annot_age_months_est_min_text_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api( '/api/annot/age/months/max/text/', methods=['GET'], __api_plural_check__=False ) def get_annot_age_months_est_max_texts(ibs, aid_list, eager=True, nInput=None): r""" annot_age_months_est_max_texts_list <- annot.annot_age_months_est_max_texts[aid_list] gets string versions of the annotation's native max age in months Args: aid_list (list): Returns: list: annot_age_months_est_max_list RESTful: Method: GET URL: /api/annot/age/months/max/text/ """ annot_age_months_est_max_list = ibs.get_annot_age_months_est_max(aid_list) annot_age_months_est_max_text_list = [ 'Unknown' if age_max in [None, -1] else '%d Months' % (age_max,) for age_max in annot_age_months_est_max_list ] return annot_age_months_est_max_text_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/age/months/text/', methods=['GET'], __api_plural_check__=False) def get_annot_age_months_est_texts(ibs, aid_list, eager=True, nInput=None): r""" annot_age_months_est_texts_list <- annot.annot_age_months_est_texts[aid_list] gets string versions of the annotation's native combined age in months Args: aid_list (list): Returns: list: annot_age_months_est_text_list RESTful: Method: GET URL: /api/annot/age/months/text/ """ annot_age_months_est_min_text_list = ibs.get_annot_age_months_est_min_texts(aid_list) annot_age_months_est_max_text_list = ibs.get_annot_age_months_est_max_texts(aid_list) annot_age_months_est_text_list = [ '%s to %s' % (age_min_text, age_max_text) for age_min_text, age_max_text in zip( annot_age_months_est_min_text_list, annot_age_months_est_max_text_list ) ] annot_age_months_est_text_list = [ 'UNKNOWN AGE' if annot_age_months_est_text == 'Unknown to Unknown' else annot_age_months_est_text for annot_age_months_est_text in annot_age_months_est_text_list ] return annot_age_months_est_text_list
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/age/months/min/', methods=['PUT'], __api_plural_check__=False) def set_annot_age_months_est_min( ibs, aid_list, annot_age_months_est_min_list, duplicate_behavior='error' ): r""" annot_age_months_est_min_list -> annot.annot_age_months_est_min[aid_list] Args: aid_list annot_age_months_est_min_list TemplateInfo: Tsetter_native_column tbl = annot col = annot_age_months_est_min RESTful: Method: PUT URL: /api/annot/age/months/min/ """ id_iter = aid_list colnames = (ANNOT_AGE_MONTHS_EST_MIN,) ibs.db.set( const.ANNOTATION_TABLE, colnames, annot_age_months_est_min_list, id_iter, duplicate_behavior=duplicate_behavior, )
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/age/months/max/', methods=['PUT'], __api_plural_check__=False) def set_annot_age_months_est_max( ibs, aid_list, annot_age_months_est_max_list, duplicate_behavior='error' ): r""" annot_age_months_est_max_list -> annot.annot_age_months_est_max[aid_list] Args: aid_list annot_age_months_est_max_list TemplateInfo: Tsetter_native_column tbl = annot col = annot_age_months_est_max RESTful: Method: PUT URL: /api/annot/age/months/max/ """ id_iter = aid_list colnames = (ANNOT_AGE_MONTHS_EST_MAX,) ibs.db.set( const.ANNOTATION_TABLE, colnames, annot_age_months_est_max_list, id_iter, duplicate_behavior=duplicate_behavior, )
[docs]@register_ibs_method @accessor_decors.getter @register_api('/api/annot/image/contributor/tag/', methods=['GET']) def get_annot_image_contributor_tag(ibs, aid_list): r""" Auto-docstr for 'get_annot_image_contributor_tag' """ gid_list = ibs.get_annot_gids(aid_list) contributor_tag_list = ibs.get_image_contributor_tag(gid_list) return contributor_tag_list
[docs]@register_ibs_method @accessor_decors.getter @register_api('/api/annot/imageset/text/', methods=['GET']) def get_annot_image_set_texts(ibs, aid_list): r""" Auto-docstr for 'get_annot_image_contributor_tag' RESTful: Method: GET URL: /api/annot/imageset/text/ """ gid_list = ibs.get_annot_gids(aid_list) imagesettext_list = ibs.get_image_imagesettext(gid_list) filter_imageset_set = set(['*Exemplars', '*All Images', '*Ungrouped Images']) filtered_imagesettext_list = [ [imageset for imageset in imagesettext if imageset not in filter_imageset_set] for imagesettext in imagesettext_list ] imagesettext_list = [ ','.join(map(str, imagesettext)) for imagesettext in filtered_imagesettext_list ] return imagesettext_list
[docs]@register_ibs_method @accessor_decors.getter def get_annot_rowids_from_partial_vuuids(ibs, partial_vuuid_strs): r""" Args: ibs (IBEISController): wbia controller object partial_uuid_list (list): CommandLine: python -m wbia.control.manual_annot_funcs --test-get_annots_from_partial_uuids Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[::2] >>> vuuids = ibs.get_annot_visual_uuids(aid_list) >>> partial_vuuid_strs = [u[0:4] for u in map(str, vuuids)] >>> aids_list = get_annot_rowids_from_partial_vuuids(ibs, partial_uuid_list) >>> print(result) [[1], [3], [5], [7], [9], [11], [13]] """ # Hackyway because I can't figure out startswith in sqlite for UUID blobs # just need to figure out how to convert a string into its corresponding byte # value, then I can do it with the referecence provided # References: # # search like with blobs # http://sqlite.1065341.n5.nabble.com/LIKE-with-BLOB-td48050.html aid_list = ibs.get_valid_aids() vuuids = ibs.get_annot_visual_uuids(aid_list) vuuid_strs = [_.replace('-', '') for _ in map(str, vuuids)] aids_list = [ [ aid for aid, vuuid in zip(aid_list, vuuid_strs) if vuuid.startswith(partial_vuuid) ] for partial_vuuid in partial_vuuid_strs ] return aids_list
# partial_vuuid = partial_vuuid_strs[0] # res = ibs.db.cur.execute( # ''' # SELECT annot_rowid from ANNOTATIONS # WHERE annot_visual_uuid LIKE ? || '%' # ''', (bytes(partial_vuuid),)) # logger.info(res.fetchall()) # res = ibs.db.cur.execute( # ''' # SELECT annot_rowid from ANNOTATIONS # WHERE annot_visual_uuid LIKE ? || '%' # ''', (partial_vuuid,)) # logger.info(res.fetchall()) # res = ibs.db.cur.execute( # ''' # SELECT annot_rowid from ANNOTATIONS # WHERE annot_visual_uuid LIKE ? || '%' # ''', (bytes(partial_vuuid),)) # logger.info(res.fetchall()) # # || - is used to concat strings # res = ibs.db.cur.execute( # ''' # SELECT annot_rowid from ANNOTATIONS # WHERE annot_note LIKE ? || '%' # ''', ('very',)) # logger.info(res.fetchall()) # pass
[docs]@register_ibs_method @accessor_decors.getter_1to1 def get_annot_tag_text(ibs, aid_list, eager=True, nInput=None): r"""annot_tags_list <- annot.annot_tags[aid_list] gets data from the "native" column "annot_tags" in the "annot" table Args: aid_list (list): Returns: list: annot_tags_list Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> ibs, config2_ = testdata_ibs() >>> aid_list = ibs._get_all_aids() >>> eager = True >>> annot_tags_list = ibs.get_annot_tag_text(aid_list, eager=eager) >>> assert len(aid_list) == len(annot_tags_list) """ id_iter = aid_list colnames = (ANNOT_TAG_TEXT,) annot_tags_list = ibs.db.get( const.ANNOTATION_TABLE, colnames, id_iter, id_colname='rowid', eager=eager, nInput=nInput, ) return annot_tags_list
[docs]@register_ibs_method @accessor_decors.setter def set_annot_tag_text(ibs, aid_list, annot_tags_list, duplicate_behavior='error'): r"""annot_tags_list -> annot.annot_tags[aid_list] Args: aid_list annot_tags_list """ # logger.info('[ibs] set_annot_tag_text of aids=%r to tags=%r' % (aid_list, annot_tags_list)) id_iter = aid_list colnames = (ANNOT_TAG_TEXT,) ibs.db.set( const.ANNOTATION_TABLE, colnames, annot_tags_list, id_iter, duplicate_behavior=duplicate_behavior, )
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/reviewed/', methods=['GET']) def get_annot_reviewed(ibs, aid_list): r""" Returns: list_ (list): "All Instances Found" flag, true if all objects of interest (animals) have an ANNOTATION in the annot RESTful: Method: GET URL: /api/annot/reviewed/ """ reviewed_list = ibs.db.get( const.ANNOTATION_TABLE, ('annot_toggle_reviewed',), aid_list ) return reviewed_list
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/reviewed/', methods=['PUT']) def set_annot_reviewed(ibs, aid_list, reviewed_list): r""" Sets the annot all instances found bit RESTful: Method: PUT URL: /api/annot/reviewed/ """ id_iter = ((aid,) for aid in aid_list) val_list = ((reviewed,) for reviewed in reviewed_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_toggle_reviewed',), val_list, id_iter)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/multiple/', methods=['GET']) def get_annot_multiple(ibs, aid_list): r""" RESTful: Method: GET URL: /api/annot/multiple/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ibs.get_valid_aids() >>> flag_list = get_annot_multiple(ibs, aid_list) >>> result = ('flag_list = %s' % (ut.repr2(flag_list),)) >>> print(result) """ flag_list = ibs.db.get(const.ANNOTATION_TABLE, ('annot_toggle_multiple',), aid_list) flag_list = [None if flag is None else bool(flag) for flag in flag_list] return flag_list
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/multiple/', methods=['PUT']) def set_annot_multiple(ibs, aid_list, flag_list): r""" Sets the annot all instances found bit RESTful: Method: PUT URL: /api/annot/multiple/ """ id_iter = ((aid,) for aid in aid_list) val_list = ((flag,) for flag in flag_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_toggle_multiple',), val_list, id_iter)
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/interest/', methods=['GET']) def get_annot_interest(ibs, aid_list): r""" RESTful: Method: GET URL: /api/annot/interest/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ibs.get_valid_aids() >>> flag_list = get_annot_interest(ibs, aid_list) >>> result = ('flag_list = %s' % (ut.repr2(flag_list),)) >>> print(result) """ flag_list = ibs.db.get(const.ANNOTATION_TABLE, ('annot_toggle_interest',), aid_list) flag_list = [None if flag is None else bool(flag) for flag in flag_list] return flag_list
[docs]@register_ibs_method @accessor_decors.getter_1to1 @register_api('/api/annot/canonical/', methods=['GET']) def get_annot_canonical(ibs, aid_list, default_none_to_false=True): r""" RESTful: Method: GET URL: /api/annot/canonical/ Example: >>> # DISABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> ibs = wbia.opendb(defaultdb='testdb1') >>> aid_list = ibs.get_valid_aids() >>> flag_list = get_annot_canonical(ibs, aid_list) >>> result = ('flag_list = %s' % (ut.repr2(flag_list),)) >>> print(result) """ try: flag_list = ibs.db.get( const.ANNOTATION_TABLE, ('annot_toggle_canonical',), aid_list ) except KeyError: flag_list = [False] * len(aid_list) if default_none_to_false: flag_list = [False if flag is None else bool(flag) for flag in flag_list] return flag_list
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/interest/', methods=['PUT']) def set_annot_interest( ibs, aid_list, flag_list, quiet_delete_thumbs=False, delete_thumbs=True ): r""" Sets the annot all instances found bit RESTful: Method: PUT URL: /api/annot/interest/ """ id_iter = ((aid,) for aid in aid_list) val_list = ((flag,) for flag in flag_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_toggle_interest',), val_list, id_iter) if delete_thumbs: gid_list = list(set(ibs.get_annot_gids(aid_list))) config2_ = {'thumbsize': 221} ibs.delete_image_thumbs(gid_list, quiet=quiet_delete_thumbs, **config2_)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/canonical/', methods=['PUT']) def set_annot_canonical(ibs, aid_list, flag_list): r""" Sets the annot all instances found bit RESTful: Method: PUT URL: /api/annot/canonical/ """ id_iter = ((aid,) for aid in aid_list) val_list = ((flag,) for flag in flag_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_toggle_canonical',), val_list, id_iter)
[docs]@register_ibs_method @register_api('/api/annot/encounter/static/', methods=['PUT']) def set_annot_static_encounter(ibs, aids, vals): id_iter = zip(aids) val_iter = zip(vals) ibs.db.set(const.ANNOTATION_TABLE, ('annot_static_encounter',), val_iter, id_iter)
[docs]@register_ibs_method @register_api('/api/annot/encounter/static/', methods=['GET']) def get_annot_static_encounter(ibs, aids): return ibs.db.get(const.ANNOTATION_TABLE, ('annot_static_encounter',), aids)
@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator(const.ANNOTATION_TABLE, [ANNOT_STAGED_FLAG], rowidx=0) @register_api('/api/annot/staged/', methods=['PUT']) def _set_annot_staged_flags(ibs, aid_list, flag_list): r""" Sets if an annotation is staged RESTful: Method: PUT URL: /api/annot/staged/ """ id_iter = ((aid,) for aid in aid_list) val_iter = ((flag,) for flag in flag_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_STAGED_FLAG,), val_iter, id_iter)
[docs]@register_ibs_method @register_api('/api/annot/staged/uuid/', methods=['PUT']) def set_annot_staged_uuids(ibs, aid_list, annot_uuid_list): r""" Returns: list_ (list): all nids of known animals (does not include unknown names) """ id_iter = ((aid,) for aid in aid_list) val_iter = ((annot_uuid,) for annot_uuid in annot_uuid_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_staged_uuid',), val_iter, id_iter) flag_list = [annot_uuid is not None for annot_uuid in annot_uuid_list] ibs._set_annot_staged_flags(aid_list, flag_list)
[docs]@register_ibs_method @accessor_decors.setter @accessor_decors.cache_invalidator( const.ANNOTATION_TABLE, [ANNOT_STAGED_USER_ID], rowidx=0 ) @register_api('/api/annot/staged/user/', methods=['PUT']) def set_annot_staged_user_ids(ibs, aid_list, user_id_list): r""" Sets the staged annotation user id RESTful: Method: PUT URL: /api/annot/staged/user/ """ id_iter = ((aid,) for aid in aid_list) val_iter = ((user_id,) for user_id in user_id_list) ibs.db.set(const.ANNOTATION_TABLE, (ANNOT_STAGED_USER_ID,), val_iter, id_iter)
[docs]@register_ibs_method @accessor_decors.setter @register_api('/api/annot/staged/metadata/', methods=['PUT']) def set_annot_staged_metadata(ibs, aid_list, metadata_dict_list): r""" Sets the annot's staged metadata using a metadata dictionary RESTful: Method: PUT URL: /api/annot/staged/metadata/ CommandLine: python -m wbia.control.manual_annot_funcs --test-set_annot_metadata Example: >>> # ENABLE_DOCTEST >>> from wbia.control.manual_annot_funcs import * # NOQA >>> import wbia >>> import random >>> # build test data >>> ibs = wbia.opendb('testdb1') >>> aid_list = ibs.get_valid_aids()[0:1] >>> metadata_dict_list = [ >>> {'test': random.uniform(0.0, 1.0)}, >>> ] >>> print(ut.repr2(metadata_dict_list)) >>> ibs.set_annot_metadata(aid_list, metadata_dict_list) >>> # verify results >>> metadata_dict_list_ = ibs.get_annot_metadata(aid_list) >>> print(ut.repr2(metadata_dict_list_)) >>> assert metadata_dict_list == metadata_dict_list_ >>> metadata_str_list = [ut.to_json(metadata_dict) for metadata_dict in metadata_dict_list] >>> print(ut.repr2(metadata_str_list)) >>> metadata_str_list_ = ibs.get_annot_metadata(aid_list, return_raw=True) >>> print(ut.repr2(metadata_str_list_)) >>> assert metadata_str_list == metadata_str_list_ """ id_iter = ((aid,) for aid in aid_list) metadata_str_list = [] for metadata_dict in metadata_dict_list: metadata_str = ut.to_json(metadata_dict) metadata_str_list.append(metadata_str) val_list = ((metadata_str,) for metadata_str in metadata_str_list) ibs.db.set(const.ANNOTATION_TABLE, ('annot_staged_metadata_json',), val_list, id_iter)
# ========== # Testdata # ==========
[docs]def testdata_ibs(): r""" Auto-docstr for 'testdata_ibs' """ import wbia ibs = wbia.opendb('testdb1') qreq_ = None return ibs, qreq_