WillowObject

class src.dataset.willow_obj.WillowObject(sets, obj_resize)

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

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 :return: (list of data, list of permutation matrices)

get_pair(cls=None, shuffle=True)

Randomly get a pair of objects from WILLOW-object dataset :param cls: None for random class, or specify for a certain set :param shuffle: random shuffle the keypoints :return: (pair of data, groundtruth permutation matrix)

len(cls)