IMC_PT_SparseGM

class src.dataset.imc_pt_sparsegm.IMC_PT_SparseGM(sets, obj_resize)

get_multi(cls=None, num=2, shuffle=True, filter_outlier=False)

Randomly get multiple objects from Willow Object Class dataset for multi-matching. :param cls: None for random class, or specify for a certain set :param num: number of objects to be fetched :param shuffle: random shuffle the keypoints :param filter_outlier: filter out outlier keypoints among images :return: (list of data, list of permutation matrices)

get_pair(cls=None, shuffle=True, tgt_outlier=False, src_outlier=False)

Randomly get a pair of objects from Photo Tourism dataset :param cls: None for random class, or specify for a certain set :param shuffle: random shuffle the keypoints :param src_outlier: allow outlier in the source graph (first graph) :param tgt_outlier: allow outlier in the target graph (second graph) :return: (pair of data, groundtruth permutation matrix)

len(cls)