use argparser for project selection
This commit is contained in:
+10
-2
@@ -1,6 +1,7 @@
|
|||||||
import os.path
|
import os.path
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import cv2 as cv
|
import cv2 as cv
|
||||||
|
from argparse import ArgumentParser
|
||||||
from stack_factory import StackFactory
|
from stack_factory import StackFactory
|
||||||
from status import Status
|
from status import Status
|
||||||
from stack_deep import StackDeep
|
from stack_deep import StackDeep
|
||||||
@@ -90,8 +91,15 @@ def main(prj_name: str = "test"):
|
|||||||
stack.state_save()
|
stack.state_save()
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# main("norb_small_16h_v2")
|
ap = ArgumentParser()
|
||||||
main("mnist_2")
|
ap.add_argument("name", type=str,
|
||||||
|
default='default',
|
||||||
|
help="Name of project")
|
||||||
|
|
||||||
|
args = ap.parse_args()
|
||||||
|
var_args = vars(args)
|
||||||
|
|
||||||
|
main(var_args["name"])
|
||||||
cv.destroyAllWindows()
|
cv.destroyAllWindows()
|
||||||
|
|
||||||
print("Test: [passed]")
|
print("Test: [passed]")
|
||||||
|
|||||||
Reference in New Issue
Block a user