예제 #1
0
파일: smk_debug.py 프로젝트: Erotemic/ibeis
def testdata_nonagg_rvec():
    from ibeis.algo.hots.smk import smk_debug
    ibs, annots_df, daids, qaids, invindex, wx2_idxs, qparams = smk_debug.testdata_raw_internals1()
    words     = invindex.words
    idx2_vec  = invindex.idx2_dvec
    wx2_dmaws  = invindex.wx2_dmaws
    idx2_daid  = invindex.idx2_daid
    # wx_sublist denotes which words are used
    wx_sublist = np.array(list(wx2_idxs.keys()))
    idxs_list  = [wx2_idxs[wx].astype(np.int32) for wx in wx_sublist]
    maws_list  = [wx2_dmaws[wx] for wx in wx_sublist]
    aids_list  = [idx2_daid.take(idxs) for idxs in idxs_list]
    return words, wx_sublist, aids_list, idxs_list, idx2_vec, maws_list
예제 #2
0
def testdata_nonagg_rvec():
    from ibeis.algo.hots.smk import smk_debug
    ibs, annots_df, daids, qaids, invindex, wx2_idxs, qparams = smk_debug.testdata_raw_internals1(
    )
    words = invindex.words
    idx2_vec = invindex.idx2_dvec
    wx2_dmaws = invindex.wx2_dmaws
    idx2_daid = invindex.idx2_daid
    # wx_sublist denotes which words are used
    wx_sublist = np.array(list(wx2_idxs.keys()))
    idxs_list = [wx2_idxs[wx].astype(np.int32) for wx in wx_sublist]
    maws_list = [wx2_dmaws[wx] for wx in wx_sublist]
    aids_list = [idx2_daid.take(idxs) for idxs in idxs_list]
    return words, wx_sublist, aids_list, idxs_list, idx2_vec, maws_list
예제 #3
0
파일: smk_debug.py 프로젝트: Erotemic/ibeis
def testdata_raw_internals1_5(**kwargs):
    """
    contains internal data up to idf weights

    Example:
        >>> from ibeis.algo.hots.smk.smk_debug import *  # NOQA
    """
    from ibeis.algo.hots.smk import smk_debug
    ibs, annots_df, daids, qaids, invindex, wx2_idxs, qparams = smk_debug.testdata_raw_internals1(**kwargs)
    print('[smk_debug] testdata_raw_internals1_5')
    words     = invindex.words
    wx_series = np.arange(len(words))
    idx2_aid  = invindex.idx2_daid
    wx2_idf = smk_index.compute_word_idf_(wx_series, wx2_idxs, idx2_aid, daids)
    invindex.wx2_idf = wx2_idf
    return ibs, annots_df, daids, qaids, invindex, wx2_idxs, qparams
예제 #4
0
def testdata_raw_internals1_5(**kwargs):
    """
    contains internal data up to idf weights

    Example:
        >>> from ibeis.algo.hots.smk.smk_debug import *  # NOQA
    """
    from ibeis.algo.hots.smk import smk_debug
    ibs, annots_df, daids, qaids, invindex, wx2_idxs, qparams = smk_debug.testdata_raw_internals1(
        **kwargs)
    print('[smk_debug] testdata_raw_internals1_5')
    words = invindex.words
    wx_series = np.arange(len(words))
    idx2_aid = invindex.idx2_daid
    wx2_idf = smk_index.compute_word_idf_(wx_series, wx2_idxs, idx2_aid, daids)
    invindex.wx2_idf = wx2_idf
    return ibs, annots_df, daids, qaids, invindex, wx2_idxs, qparams