cvtools.label_convert.dota_to_coco module

class cvtools.label_convert.dota_to_coco.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')[源代码]