kpick.matching.qatm package

Submodules

kpick.matching.qatm.QATMer module

class kpick.matching.qatm.QATMer.QATMER

Bases: object

get_boxes(im, tmp_name=None)

get_model(tmp_dir, data_mean, data_std, alpha=25, hard_thresh=0.025, use_cuda=True)

get_scores(im, tmp_name=None)

class kpick.matching.qatm.QATMer.TemplateDataset(tmp_dir, transform): Bases: Dataset

kpick.matching.qatm.QATMer.demo_qatm_gui(title='QATM GUI', Detector=<class 'kpick.matching.qatm.QATMer.QATMER'>, cfg_path='kpick/matching/qatm/test_data/configs/qatm.cfg', default_cfg_path='kpick/matching/qatm/test_data/configs/default.cfg', sensors=['realsense'], key_args=[], data_root='kpick/matching/qatm/test_data/samples', rgb_formats=['*'], depth_formats=None)

kpick.matching.qatm.QATMer.get_qatm_default_config()

kpick.matching.qatm.QATMer.get_qatm_gui(Detectorobj=<class 'kpick.matching.qatm.QATMer.get_qatm_obj.<locals>.QATMObj'>)

kpick.matching.qatm.QATMer.get_qatm_obj(Detector=<class 'kpick.matching.qatm.QATMer.QATMER'>)

kpick.matching.qatm.QATMer.place_on_background(im, bg)

kpick.matching.qatm.QATMer.test_qatmer()

kpick.matching.qatm.qatm module

class kpick.matching.qatm.qatm.CreateModel(alpha, model, use_cuda): Bases: object

class kpick.matching.qatm.qatm.Featex(model, use_cuda)

Bases: object

save_feature1(module, input, output)

save_feature2(module, input, output)

class kpick.matching.qatm.qatm.ImageDataset(template_dir_path, image_name, thresh_csv=None, transform=None): Bases: Dataset

class kpick.matching.qatm.qatm.MyNormLayer: Bases: object

class kpick.matching.qatm.qatm.QATM(alpha)

Bases: object

compute_output_shape(input_shape)

kpick.matching.qatm.qatm.nms(score, w_ini, h_ini, thresh=0.7)

kpick.matching.qatm.qatm.nms_multi(scores, w_array, h_array, thresh_list)

kpick.matching.qatm.qatm.plot_result(image_raw, boxes, show=False, save_name=None, color=(255, 0, 0))

kpick.matching.qatm.qatm.plot_result_multi(image_raw, boxes, indices, show=False, save_name=None, color_list=None)

kpick.matching.qatm.qatm.run_multi_sample(model, dataset)

kpick.matching.qatm.qatm.run_one_sample(model, template, image, image_name='unnamed')

kpick.matching.qatm.utils module

kpick.matching.qatm.utils.IoU(r1, r2)

kpick.matching.qatm.utils.all_sample_iou(score_list, gt_list)

kpick.matching.qatm.utils.compute_score(x, w, h)

kpick.matching.qatm.utils.evaluate_iou(rect_gt, rect_pred)

kpick.matching.qatm.utils.int_(x)

kpick.matching.qatm.utils.locate_bbox(a, w, h)

kpick.matching.qatm.utils.plot_success_curve(iou_score, title='')

kpick.matching.qatm.utils.score2curve(score, thres_delta=0.01)

Module contents