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

Update labelme2yolo.py

This commit is contained in:
rooneysh 2021-08-18 15:44:58 +08:00 committed by GitHub
parent 3b4082e236
commit c87828c117
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
labelme2yolo.py
View File

@ -132,15 +132,12 @@ class Labelme2YOLO(object):
return img_path
if __name__ == '__main__':
argv = sys.argv[1:]
parser = argparse.ArgumentParser()
parser.add_argument('--json_dir',type=str,
help='Please input the path of the labelme json files.')
parser.add_argument('--val_size',type=str,
parser.add_argument('--val_size',type=float,
help='Please input the validation dataset size, for example 0.1 ')
json_dir, val_size = parser.parse_args(argv)
convertor = Labelme2YOLO(json_dir)
convertor.convert(val_size=val_size)
args = parser.parse_args(sys.argv[1:])
convertor = Labelme2YOLO(args.json_dir)
convertor.convert(val_size=args.val_size)