shuoshuoshuo
/
labelme2yolo
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reformat code

This commit is contained in:
Wang Xin 2022-08-01 15:51:49 +08:00
parent 845afa0328
commit 99d963316d
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/labelme2yolo/__main__.py
View File

@ -6,4 +6,4 @@ import sys
if __name__ == '__main__':
from .cli import run
sys.exit(run())
sys.exit(run())

16
src/labelme2yolo/cli/__init__.py
View File

@ -9,23 +9,23 @@ from labelme2yolo.l2y import Labelme2YOLO
def run():
parser = argparse.ArgumentParser("labelme2yolo")
parser.add_argument('--json_dir',type=str,
parser.add_argument('--json_dir', type=str,
help='Please input the path of the labelme json files.')
parser.add_argument('--val_size',type=float, nargs='?', default=None,
parser.add_argument('--val_size', type=float, nargs='?', default=None,
help='Please input the validation dataset size, for example 0.1 ')
parser.add_argument('--test_size',type=float, nargs='?', default=0.0,
parser.add_argument('--test_size', type=float, nargs='?', default=0.0,
help='Please input the validation dataset size, for example 0.1 ')
parser.add_argument('--json_name',type=str, nargs='?', default=None,
parser.add_argument('--json_name', type=str, nargs='?', default=None,
help='If you put json name, it would convert only one json file to YOLO.')
args = parser.parse_args()
if not args.json_dir:
parser.print_help(sys.stderr)
sys.exit(1)
convertor = Labelme2YOLO(args.json_dir)
if args.json_name is None:
convertor.convert(val_size=args.val_size, test_size=args.test_size)
else:
convertor.convert_one(args.json_name)
convertor.convert_one(args.json_name)