TODO: Add Argument parsing
--dir -> pet images folder --arch -> CNN Architecture --dogfile -> File with Dog Names
This commit is contained in:
@@ -17,6 +17,7 @@
|
||||
# Imports python modules
|
||||
import argparse
|
||||
|
||||
|
||||
# TODO 1: Define get_input_args function below please be certain to replace None
|
||||
# in the return statement with parser.parse_args() parsed argument
|
||||
# collection that you created with this function
|
||||
@@ -39,10 +40,17 @@ def get_input_args():
|
||||
parse_args() -data structure that stores the command line arguments object
|
||||
"""
|
||||
# Create Parse using ArgumentParser
|
||||
parser = argparse.ArgumentParser(description="Identify dog breeds in images")
|
||||
|
||||
# Create 3 command line arguments as mentioned above using add_argument() from ArguementParser method
|
||||
|
||||
parser.add_argument(
|
||||
"-d", "--dir", default="pet_images/", help="path to the pet images folder"
|
||||
)
|
||||
parser.add_argument("-a", "--arch", default="vgg", help="CNN Model Architecture")
|
||||
parser.add_argument(
|
||||
"-f", "--dogfile", default="dognames.txt", help="Textfile with Dog Names"
|
||||
)
|
||||
|
||||
# Replace None with parser.parse_args() parsed argument collection that
|
||||
# you created with this function
|
||||
return None
|
||||
return parser.parse_args()
|
||||
|
||||
Reference in New Issue
Block a user