diff --git a/ai-programming-with-python/C02P8-pre-trained-image-classifier/get_input_args.py b/ai-programming-with-python/C02P8-pre-trained-image-classifier/get_input_args.py index e74c961..ae937a6 100644 --- a/ai-programming-with-python/C02P8-pre-trained-image-classifier/get_input_args.py +++ b/ai-programming-with-python/C02P8-pre-trained-image-classifier/get_input_args.py @@ -1,12 +1,12 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- # */AIPND-revision/intropyproject-classify-pet-images/get_input_args.py -# -# PROGRAMMER: -# DATE CREATED: -# REVISED DATE: -# PURPOSE: Create a function that retrieves the following 3 command line inputs -# from the user using the Argparse Python module. If the user fails to +# +# PROGRAMMER: +# DATE CREATED: +# REVISED DATE: +# PURPOSE: Create a function that retrieves the following 3 command line inputs +# from the user using the Argparse Python module. If the user fails to # provide some or all of the 3 inputs, then the default values are # used for the missing inputs. Command Line Arguments: # 1. Image Folder as --dir with default value 'pet_images' @@ -17,17 +17,18 @@ # 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 +# in the return statement with parser.parse_args() parsed argument # collection that you created with this function -# +# def get_input_args(): """ Retrieves and parses the 3 command line arguments provided by the user when - they run the program from a terminal window. This function uses Python's - argparse module to created and defined these 3 command line arguments. If - the user fails to provide some or all of the 3 arguments, then the default - values are used for the missing arguments. + they run the program from a terminal window. This function uses Python's + argparse module to created and defined these 3 command line arguments. If + the user fails to provide some or all of the 3 arguments, then the default + values are used for the missing arguments. Command Line Arguments: 1. Image Folder as --dir with default value 'pet_images' 2. CNN Model Architecture as --arch with default value 'vgg' @@ -36,13 +37,20 @@ def get_input_args(): Parameters: None - simply using argparse module to create & store command line arguments Returns: - parse_args() -data structure that stores the command line arguments object + 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 - - - # Replace None with parser.parse_args() parsed argument collection that - # you created with this function - return None + 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 parser.parse_args()