def __init__(self, cfg):
self.name = cfg.VISDRONE.NAME
self.img_extension = cfg.VISDRONE.IMG_EXTENSION
self.anno_extension = cfg.VISDRONE.ANNO_EXTENSION
if cfg.ANNO.PARSER == self.name:
self.src_dir = cfg.INPUT.DIR
self.image_folder = cfg.INPUT.IMAGE_FOLDER
self.label_folder = cfg.INPUT.LABEL_FOLDER
if cfg.ANNO.CREATOR == self.name:
self.dst_dir = cfg.OUTPUT.DIR
self.image_folder = cfg.OUTPUT.IMAGE_FOLDER
self.label_folder = cfg.OUTPUT.LABEL_FOLDER
self.verbose = cfg.ANNO.VERBOSE
self.logger = setup_logger(__name__)
def __init__(self, cfg):
self.parser = build_anno(cfg.ANNO.PARSER, cfg)
self.creator = build_anno(cfg.ANNO.CREATOR, cfg)
self.logger = setup_logger(__name__)
self.verbose = cfg.ANNO.VERBOSE