apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Parse command-line arguments using argparse

Browse Source
This commit is contained in:
Laszlo Treszkai 2020-07-16 15:29:52 +02:00 committed by darrenl
parent 1a9b489bb4
commit b21b48e640
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
labelImg.py
View File

@ -1,5 +1,6 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import argparse
import codecs import codecs
import distutils.spawn import distutils.spawn
import os.path import os.path
@ -1506,12 +1507,19 @@ def get_main_app(argv=[]):
app.setApplicationName(__appname__) app.setApplicationName(__appname__)
app.setWindowIcon(newIcon("app")) app.setWindowIcon(newIcon("app"))
# Tzutalin 201705+: Accept extra agruments to change predefined class file # Tzutalin 201705+: Accept extra agruments to change predefined class file
argparser = argparse.ArgumentParser()
argparser.add_argument("image_dir", nargs="?")
argparser.add_argument("predefined_classes_file",
default=os.path.join(
os.path.dirname(argv[0]),
"data", "predefined_classes.txt"),
nargs="?")
argparser.add_argument("save_dir", nargs="?")
args = argparser.parse_args(argv[1:])
# Usage : labelImg.py image predefClassFile saveDir # Usage : labelImg.py image predefClassFile saveDir
win = MainWindow(argv[1] if len(argv) >= 2 else None, win = MainWindow(args.image_dir,
argv[2] if len(argv) >= 3 else os.path.join( args.predefined_classes_file,
os.path.dirname(sys.argv[0]), args.save_dir)
'data', 'predefined_classes.txt'),
argv[3] if len(argv) >= 4 else None)
win.show() win.show()
return app, win return app, win