def load_triplets(triplet_path, subset):
smts_api = SMTSApi(triplet_path)
triplets = smts_api.get_triplets(subset)
return triplets, make_negative_list(triplets)
def load_dataset(self, dataset_name):
self.dataset_ = SMTSApi(dataset_root=self.config_.DATASET_ROOT,
name=dataset_name)
def load_triplets_bbox(triplet_path, subset):
smts_api = SMTSApi(triplet_path)
triplets, bbox = smts_api.get_triplets_bbox(subset)
return triplets, make_negative_list_bbox(triplets, subset), bbox
