kpick.dataset package
Submodules
kpick.dataset.coco module
- class kpick.dataset.coco.CocoGui(args=None, cfg_path=None, name='unnamed', default_args=None)
Bases:
DetGuiObj- augmentation_and_save(rgb, filename='unnamed.png', sub_thread=None)
- concat_anns()
- finalize_acc()
- gui_process_single(rgbd, method_ind=0, filename='unnamed', disp_mode='rgb', **kwargs)
- init_acc()
- mask_to_coco(rgbd, filename, im_dir)
- show_ann_single(rgbd, filename, im_dir)
- class kpick.dataset.coco.CocoUtils(ann_path=None)
Bases:
object- acc_augmentation_db(rgb, masks, category_ids, sub_thread=None)
- acc_single_db(rgb, masks, category_ids, sub_thread=None)
- aug_category(category_id, angle, flip_axis)
- aug_mask(mask, angle, flip_axis)
- aug_single(im_path, im_id, bg_ims=None, angle_step=10, show_step=False)
- augmentation(im_dir, ann_path, bg_dir=None, angle_step=10, show_step=False)
- binary_mask_to_rle(binary_mask)
- filename2Id(filename)
- filename2Instances(filename)
- finalize_acc()
- get_im_from_ind(image_id, im_dir)
- get_im_path(image_id, im_dir)
- get_instance_from_ind(ind, ann_path=None)
- init_acc(save_dirs, classes, aug_args=None, supercategories=None)
- make_ann_categories(classes, kpt_labels=None, kpt_skeletons=None)
- make_ann_images(image_id, im_path, im_size, image_dict_list=[])
- make_annotations(mask, bbox, ann_id, image_id, cls_id, keypoints=None, annotations=[])
- make_categories(classes, supercategories=None)
- replace_background(im_dir, background_im, select_im_inds=None, div=(2, 1))
- save(ann_path, images=None, annotations=None, categories=None, ann_info=None, ann_license=None)
- segmToMask(segm, h, w)
- segmToRLE(segm, h, w)
Convert annotation which can be polygons, uncompressed RLE to RLE. :return: binary mask (numpy 2D array)
- show_ann_single(im, filename)
- show_ims(im_dir, title='coco_viewer', im_size=(1080, 720), start_from=0, step=1)
- show_instances(im_dir, title='coco_viewer', im_size=(1080, 720))
- split_trainval(div=(2, 1))
- split_trainval_(div=(2, 1))
- visualize_instance(rgb, instance, categories=None)
- kpick.dataset.coco.concat_cocos(ann_files, im_dirs, out_file)
- kpick.dataset.coco.replace_background(ann_file, im_dir, background_path, div=[2, 1], select_stride=None)
- kpick.dataset.coco.run_coco_augmentation(root_dir, im_dir='imgs', ann_file='ann.json')
- kpick.dataset.coco.run_coco_gui(cfg_path=None, default_cfg_path=None, data_dir=None, rgb_formats=None, depth_formats=None, key_args=[])
kpick.dataset.cornell module
- class kpick.dataset.cornell.CornellDataset
Bases:
object- get_instance(rgbFilePath, rgb_format='*r.png', depth_format='*d.tiff', neg_format='*cneg.txt', pos_format='*cpos.txt')
- get_poses(filepath)
- infer_path(src_path, src_format, dst_format)
- movePose5D(pose5D, max_trans=10, max_rot=3)
- pose2Pose5D(pose)
- pose5D2pose(pose5D)
- show_apose(im, pos, neg_pose=False, color=None)
- show_instance(rgbFilePath, rgb_format='*r.png', depth_format='*d.tiff', neg_format='*cneg.txt', pos_format='*cpos.txt')
- class kpick.dataset.cornell.CornellDatasetGui(args=None, cfg_path=None, name='unnamed', default_args=None)
Bases:
CornellDataset,DetGuiObj- gui_process_single(rgbd, method_ind=0, filename='unnamed', disp_mode='rgb', **kwargs)
- show_poses(rgbd, filename='unnamed.png', disp_mode='rgb', im_dir='')
- show_positive_poses(rgbd, filename='unnamed.png', disp_mode='rgb', im_dir='')
- kpick.dataset.cornell.demo_cornell_gui(cfg_path=None)
- kpick.dataset.cornell.get_default_configs()
- kpick.dataset.cornell.test()
kpick.dataset.grip_via_dataset module
- class kpick.dataset.grip_via_dataset.GripViaAnnotation
Bases:
ViaAnnotation- draw_region(im, X, Y, disp_args, color=(0, 255, 0))
- finalize_acc()
- init(ann_json, im_dir, input_shape, classes=None)
- init_acc()
- make_db(rgbd, filename, dataset_args, net_args, disp_args, disp_mode='rgb')
- to_train(count, train_val_div)
- class kpick.dataset.grip_via_dataset.GripViaAnnotationGuiObj(args=None, cfg_path=None, name='unnamed', default_args=None)
Bases:
GripViaAnnotation,DetGuiObj- gui_process_single(rgbd, method_ind=0, filename='unnamed', disp_mode='rgb', **kwargs)
- kpick.dataset.grip_via_dataset.demo_grip_via_annocation_gui(cfg_path='configs/grip_via_annotation.cfg', default_cfg_path='configs/default_via_annotation.cfg', data_root=None, rgb_formats=None, depth_formats=None)
- kpick.dataset.grip_via_dataset.get_grip_via_ann_default_args()
kpick.dataset.via_dataset module
- class kpick.dataset.via_dataset.ViaAnnotation
Bases:
object- draw_region(im, X, Y, disp_args, color=(0, 255, 0))
- get_regions(filename)
- init(ann_json, im_dir, classes=None)
- show_single_ann(rgbd, filename, disp_args, disp_mode='rgb')
- class kpick.dataset.via_dataset.ViaAnnotationGuiObj(args=None, cfg_path=None, name='unnamed', default_args=None)
Bases:
ViaAnnotation,DetGuiObj- gui_process_single(rgbd, method_ind=0, filename='unnamed', disp_mode='rgb', **kwargs)
- kpick.dataset.via_dataset.demo_via_annocation_gui(cfg_path='configs/via_annotation.cfg', default_cfg_path='configs/default_via_annotation.cfg', data_root=None, rgb_formats=None, depth_formats=None)
- kpick.dataset.via_dataset.get_via_annotation_default_args()