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