kpick.object_detection package
Submodules
kpick.object_detection.demo module
kpick.object_detection.install module
- kpick.object_detection.install.install_nanodet()
kpick.object_detection.model module
kpick.object_detection.nanodet_detector module
- class kpick.object_detection.nanodet_detector.BaseNanoDet
Bases:
object- get_model(cfg_file, ckpt, thresh=0.5, device='cuda:0')
- run(im, thresh=0.5, filename='unnamed', classes=None, lefttop=(0, 0))
- class kpick.object_detection.nanodet_detector.NanoDetectorGui(args=None, cfg_path=None, name='unnamed', default_args=None)
Bases:
NanodetDetector,DetGuiObj- get_model()
get model
- gui_process_single(rgbd, method_ind=0, filename='unnamed', disp_mode='rgb', detected=None)
- show_det(im, ret)
- class kpick.object_detection.nanodet_detector.NanodetDetector
Bases:
BaseDetector,BaseNanoDet- get_model(cfg_file, ckpt, thresh=0.5, device='cuda:0')
- run(im, thresh=0.5, filename='unnamed', classes=None, lefttop=(0, 0))
- kpick.object_detection.nanodet_detector.test_nanodet_gui(cfg_file, ckpt, score_thresh=0.5, class_names=['unnamed'], run_realsense=False, device='cuda:0')
kpick.object_detection.test module
- class kpick.object_detection.test.Predictor(cfg, model_path, logger, device='cuda:0')
Bases:
object- inference(img)
- visualize(dets, meta, class_names, score_thres, wait=0)
- kpick.object_detection.test.get_image_list(path)
- kpick.object_detection.test.main()
- kpick.object_detection.test.parse_args()