kpick package
Subpackages
- kpick.apps package
- Subpackages
- kpick.apps.dataset package
- kpick.apps.kimchi package
- kpick.apps.mujin package
- kpick.apps.orderpicking package
- kpick.apps.robotplus package
- Subpackages
- Submodules
- kpick.apps.robotplus.copper_detector module
- kpick.apps.robotplus.detector module
- kpick.apps.robotplus.halcon_pose_estimator module
- kpick.apps.robotplus.oi_detector module
- kpick.apps.robotplus.small_copper_detector module
- kpick.apps.robotplus.test_matching2d module
- Module contents
- kpick.apps.samples package
- Submodules
- kpick.apps.samples.demo_client_gui module
- kpick.apps.samples.demo_grip module
- kpick.apps.samples.demo_module_combine module
- kpick.apps.samples.demo_picking module
- kpick.apps.samples.demo_suction module
- kpick.apps.samples.run_sensors module
- kpick.apps.samples.select_sample module
- Module contents
- Submodules
- kpick.apps.test module
- Module contents
- Subpackages
- kpick.base package
- kpick.classifier package
- Subpackages
- kpick.classifier.cifar_classification_models package
- Submodules
- kpick.classifier.cifar_classification_models.alexnet module
- kpick.classifier.cifar_classification_models.densenet module
- kpick.classifier.cifar_classification_models.preresnet module
- kpick.classifier.cifar_classification_models.resnet module
- kpick.classifier.cifar_classification_models.resnet_roi module
- kpick.classifier.cifar_classification_models.resnext module
- kpick.classifier.cifar_classification_models.vgg module
- kpick.classifier.cifar_classification_models.wrn module
- Module contents
- kpick.classifier.classification_utils package
- kpick.classifier.models package
- kpick.classifier.cifar_classification_models package
- Submodules
- kpick.classifier.basic_classification_models module
- kpick.classifier.classifier_base module
- kpick.classifier.classifier_base_org module
- kpick.classifier.roi_classifier module
- Module contents
- Subpackages
- kpick.dataset package
- kpick.instance_segmentation package
- kpick.matching package
- kpick.object_detection package
- kpick.pick package
- kpick.pose_estimation package
- kpick.segmentation package
Submodules
kpick.detector_base module
- class kpick.detector_base.BaseDetector
Bases:
object- get_model(**kwargs)
- predict_array(anArray, model, need_rot=False, test_shape=None, lefttop=(0, 0), classes=None, thresh=None, sort=None)
- run(im, model, thresh=None, classes=None)
- show_boxes(im, boxes, labels=None, scores=None, colors=None, line_thick=2, text_scale=1, text_thick=2)
- show_keypoint_inst(im, kpts, scores=None, skeleton=None, thresh=0.5, marker_size=5, text_scale=1)
- show_keypoints(im, keypoints, skeleton=None, thresh=0.5, marker_size=5, text_scale=1)
- show_mask(im, masks, colors=None)
- show_predict(im, predict, kpt_skeleton=None, colors=None, line_thick=2, text_scale=1, text_thick=2, marker_size=5)
- uncrop_predict(predict, im_size, bbox)
kpick.detectron2_detector module
- class kpick.detectron2_detector.BaseDetectron
Bases:
object- get_model(model_cfg_path, model_path, num_classes=1, score_thresh=0.5)
- run(im, model, thresh=None, classes=None)
- run_extract(predictions)
- class kpick.detectron2_detector.DetectronDetector
Bases:
BaseDetector,BaseDetectron- get_model(model_cfg_path, model_path, num_classes=1, score_thresh=0.5)
- predict_single(rgbd, model, net_args=None)
- run(im, model, thresh=None, classes=None)
- kpick.detectron2_detector.demo_detectron_gui(model_cfg_path=None, model_path=None, num_classes=None, cfg_path=None, default_cfg_path=None, DetectorGuiObj=<class 'kpick.detectron2_detector.get_detectron_gui_obj.<locals>.DetectronDetectorGuiObj'>, score_thresh=0.5, kpt_skeleton=None, key_args=None, sensors=['realsense'], data_root=None, rgb_formats=None, depth_formats=None)
- kpick.detectron2_detector.get_detectron2_default_args()
- kpick.detectron2_detector.get_detectron_gui_obj(DetectorObj=<class 'kpick.detectron2_detector.get_detectron_obj.<locals>.DetectronDetectorObj'>)
- kpick.detectron2_detector.get_detectron_obj(Detector=<class 'kpick.detectron2_detector.DetectronDetector'>)