def importDataset(self, args: Arg):
self.projectsPath.mkdir(exist_ok=True, parents=True)
self.projectsPathListFromDataset = [
self.projectsPath / path.name
for path in self.getSubDirList(self.datasetsPath)
]
for datasetPath in self.datasetPathList:
projPath = self.projectsPath / datasetPath.name
projImportArgs = [
'project', 'import', '-i',
str(datasetPath), '-o',
str(projPath), '-f', args['format'].lower(), '--overwrite'
]
main(projImportArgs)
return self
def exportDataset(self, args: Arg, merge=False):
if merge:
projectsPathList = [self.projectsPath / self.mergeFolderName]
else:
if not hasattr(self, 'projectsPathListFromDataset'):
importArgs = ImportArg()
self.importDataset(importArgs)
projectsPathList = self.projectsPathListFromDataset
for proj in projectsPathList:
exportPath = (self.projectsPath / '..' / 'export' /
f'{proj.name}_{args["format"].lower()}').absolute()
if exportPath.is_dir():
rmtree(exportPath, onerror=remove_readonly)
exportPath.mkdir(exist_ok=True, parents=True)
export_args = [
'project', 'export', '-o',
str(exportPath), '-p',
str(proj), '-f', args['format'].lower(), '--', '--save-images'
]
main(export_args)
return self
#!/usr/bin/env python
import sys
from datumaro.cli.__main__ import main
if __name__ == '__main__':
sys.exit(main())
def run_datum(test, *args, expected_code=0):
from datumaro.cli.__main__ import main
test.assertEqual(expected_code, main(args), str(args))