class AFWDarknetTests(unittest.TestCase):
def setUp(self):
rmtree(TMP_DIR, ignore_errors=True)
self.afw = AFW(AFW_DIR)
def test_sanity(self):
self.afw.export(TMP_DIR, 'darknet')
class CocoExporterTests(unittest.TestCase):
def setUp(self):
rmtree(TMP_DIR, ignore_errors=True)
self.afw = AFW(AFW_DIR)
def test_sanity(self):
self.afw.export(TMP_DIR, 'coco')
class AFWTests(unittest.TestCase):
def setUp(self):
self.afw = AFW(AFW_DIR)
def test_train_set(self):
for i in self.afw.images():
log.debug(i)
for f in i.faces:
log.debug(f)
@unittest.skip("skipping because it takes too long")
def test_download(self):
self.afw = AFW(os.path.join(TMP_DIR, 'afw_download/'))
self.afw.download()
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)