cvtools.label_convert package

Subpackages

• cvtools.label_convert.arcsoft package
  • Submodules
    • cvtools.label_convert.arcsoft.attribute_combination module
  • Module contents
• cvtools.label_convert.coco package
  • Submodules
    • cvtools.label_convert.coco.merge_coco module
    • cvtools.label_convert.coco.split_coco module
  • Module contents

Submodules

• cvtools.label_convert.arcsoft_to_coco module
• cvtools.label_convert.bdd_to_coco module
• cvtools.label_convert.coco_to_dets module
• cvtools.label_convert.coco_to_jiang module
• cvtools.label_convert.dota_to_coco module
• cvtools.label_convert.jiang_to_coco module
• cvtools.label_convert.voc_to_coco module
• cvtools.label_convert.voc_to_darknet module

Module contents

class cvtools.label_convert.VOC2COCO(root, mode='train', cls=['aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor'], cls_replace=None, use_xml_name=True, read_test=False)

基类：object

convert voc-like dataset to coco-like dataset

参数:

• root (str) -- path include images, xml, file list
• mode (str) -- 'train', 'val', 'trainval', 'test'. used to find file list.
• cls (str or list) -- class name in a file or a list.
• cls_replace (dict) -- a dictionary for replacing class name. if not needed, you can just ignore it.
• use_xml_name (bool) -- image filename source, if true, using the same name as xml for the image, otherwise using 'filename' in xml context for the image.
• read_test (bool) -- Test if the picture can be read normally.

check_image(img_path)

convert(filter_objs=None)

save_json(to_file='cocolike.json')

class cvtools.label_convert.VOC2DarkNet(voc_root, mode, classes=['aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor'], use_xml_name=True, read_test=False)

基类：object

check_image(img_path)

convert(save_root=None)

convert_box(size, box)

class cvtools.label_convert.DOTA2COCO(label_root, image_root, classes=['large-vehicle', 'swimming-pool', 'helicopter', 'bridge', 'plane', 'ship', 'soccer-ball-field', 'basketball-court', 'ground-track-field', 'small-vehicle', 'harbor', 'baseball-diamond', 'tennis-court', 'roundabout', 'storage-tank'], path_replace=None, box_form='x1y1wh')

基类：object

convert DOTA labels to coco-like format labels.

参数:

• label_root (str) -- label file path, for example, '/home/data/DOTA/train/labelTxt'
• image_root (str) -- image path, for example, '/home/data/DOTA/train/images'
• classes (str or list) -- class name in a file or a list.
• path_replace (dict) -- replace same things in images path, if not needed, you can just ignore it.
• box_form (str) -- coco bbox format, default 'x1y1wh'.

check_box(box, w, h)

convert(use_crop=False)

save_json(to_file='cocolike.json')

cvtools.label_convert.rect_reserved(label)

cvtools.label_convert.face_reserved(label, have_assert=False)

cvtools.label_convert.head_reserved(label, have_assert=False)

cvtools.label_convert.gender_reserved(label)

class cvtools.label_convert.COCO2Dets(anns_file, num_coors=4)

基类：object

将DOTA-COCO兼容格式GT转成检测结果表达形式results，保存成pkl results: {

image_id: dets, # image_id必须是anns中有效的id image_id: dets, ...

} dets: {

cls_id：[[位置坐标，得分], [...], ...], cls_id: [[位置坐标，得分], [...], ...], ...

}，

convert(to_file=None)

handle_ann(ann)

如果想自定义ann处理方式，继承此类，然后重新实现此方法

save_json(to_file)

save_pkl(to_file)