def create_dataset(dataset_name, data_dir): if not dataset_name: dataset_name = 'mafa' if dataset_name == 'widerface': from morghulis.widerface import Wider ds = Wider(data_dir) elif dataset_name == 'fddb': from morghulis.fddb import FDDB ds = FDDB(data_dir) elif dataset_name == 'afw': from morghulis.afw import AFW ds = AFW(data_dir) elif dataset_name == 'pascal_faces': from morghulis.pascal_faces import PascalFaces ds = PascalFaces(data_dir) elif dataset_name == 'mafa': from morghulis.mafa import Mafa ds = Mafa(data_dir) elif dataset_name == 'caltech': from morghulis.caltech_faces import CaltechFaces ds = CaltechFaces(data_dir) elif dataset_name == 'ufdd': from morghulis.ufdd import UFDD ds = UFDD(data_dir) else: logging.error('Invalid dataset name %s', dataset_name) raise ValueError('Invalid dataset name %s' % dataset_name) return ds
def main(): parser = argparse.ArgumentParser() parser.add_argument('--dataset', dest='dataset', action='store', required=True, help='widerface or fddb') parser.add_argument('--format', dest='format', action='store', required=True, help='darknet, tensorflow, caffe or coco') parser.add_argument('--data_dir', dest='data_dir', action='store', required=True, help='') parser.add_argument('--output_dir', dest='output_dir', action='store', required=True, help='') args = parser.parse_args() dataset = args.dataset _format = args.format data_dir = args.data_dir output_dir = args.output_dir if dataset == 'widerface': from morghulis.widerface import Wider ds = Wider(data_dir) elif dataset == 'fddb': from morghulis.fddb import FDDB ds = FDDB(data_dir) elif dataset == 'afw': from morghulis.afw import AFW ds = AFW(data_dir) else: logging.error('Invalid dataset name %s', dataset) ds.export(output_dir, _format)
def main(): parser = argparse.ArgumentParser() parser.add_argument('--dataset', dest='dataset', action='store', required=True, help='widerface, fddb or afw') parser.add_argument('--output_dir', dest='output_dir', action='store', required=True, help='') args = parser.parse_args() dataset = args.dataset output_dir = args.output_dir if dataset == 'widerface': from morghulis.widerface import Wider ds = Wider(output_dir) elif dataset == 'fddb': from morghulis.fddb import FDDB ds = FDDB(output_dir) elif dataset == 'afw': from morghulis.afw import AFW ds = AFW(output_dir) elif dataset == 'pascal_faces': from morghulis.pascal_faces import PascalFaces ds = PascalFaces(output_dir) else: logging.error('Invalid dataset name %s', dataset) raise ValueError('Invalid dataset name %s' % dataset) ds.download()
def test_download(self): self.afw = AFW(os.path.join(TMP_DIR, 'afw_download/')) self.afw.download()
def setUp(self): self.afw = AFW(AFW_DIR)
def setUp(self): rmtree(TMP_DIR, ignore_errors=True) self.afw = AFW(AFW_DIR)