config.py

import os
import argparse

def parse_args():
	parser = argparse.ArgumentParser()
	parser.add_argument('-d', '--darknet', dest='DARKNET', action='store_true', help="Specifies to use the Darknet classifier")
	parser.add_argument('-k', '--keras', dest='KERAS', action='store_true', help="Specifies to use the Keras classifier")
	parser.add_argument('-t', '--tesseract', dest='TESSERACT', action='store_true', help="Use the local Tesseract OCR engine")
	parser.add_argument('-c', '--cognitive_services', dest='COGNITIVE_SERVICES', action='store_true', help="Use Cognitive Servies for OCR")
	parser.add_argument('-s', '--key', dest="SUBSCRIPTION_KEY", default="", help="Subscription Key for Cognitive Services")
	parser.add_argument('-l', '--dbl', dest="DARKNET_BINARY_LOCATION", default=None, help="Location of Darknet Binary")
	parser.add_argument('--thresh', dest="DARKNET_THRESH", default=.25, type=float, help="Darknet threshold for successful classification (lower = more bounding boxes)")
	parser.add_argument('--data', dest="DARKNET_DATA_FILE", default="data/obj.data", help="Darknet data file")
	parser.add_argument('--cfg', dest="DARKNET_CFG_FILE", default="yolo-obj.cfg", help="Darknet configuration file")
	parser.add_argument('--weights', dest="DARKNET_WEIGHTS", default="yolo-obj_1600.weights", help="Weights for Darknet")
	parser.add_argument('-e', '--kl', dest="KERAS_LOCATION", default="keras-yolo3/", help="Location of Keras-yolo3")
	parser.add_argument('-r', '--show_response', dest="SHOW_RESPONSE", action='store_false', help="Shows responses from OCR")
	parser.add_argument('-i', '--show_images', dest="SHOW_IMAGES", action='store_true', help="Shows images after cropping and rotating")
	parser.add_argument('-n', '--label_name', dest='LABEL_NAME', default='label', help="Name of label for detection")
	parser.add_argument('-o', '--rotnet_location', dest='ROTNET_LOCATION', default="./RotNet", help="Location of RotNet")
	parser.add_argument('-m', '--model_name', dest='ROTNET_MODEL_NAME', default="rotnet_models/rotnet_street_view_resnet50_keras2.hdf5", help="Location of RotNet Model")
	parser.add_argument('-f', '--file_name', dest='ROTNET_SAVE_FILE_NAME', default="tilted.jpg", help="Where to save for RotNet")
	parser.add_argument('--local', dest='LOCAL_DATABASE', action='store_true', help="Use local database")
	parser.add_argument('-x', '--cosmos', dest='COSMOS_DATABASE', action='store_true', help='Use Cosmos database')
	args = parser.parse_args()
	if args.KERAS == False and args.DARKNET == False:
		parser.error("Either Darknet or Keras must be set, add -k or -d")
	if args.TESSERACT == False and args.COGNITIVE_SERVICES == False:
		parser.error("Either Tesseract or Cognitive Services must be set, add -t or -c")
	if args.COGNITIVE_SERVICES == True and args.SUBSCRIPTION_KEY == "":
		parser.error("Cognitive Services needs a subscription key, please provide with -s")
	return args


## Change the following variable based on what algorithms you want to use ##
global DARKNET, KERAS, TESSERACT, COGNITIVE_SERVICES, DARKNET_BINARY_LOCATION, DARKNET_THRESH, DARKNET_DATA_FILE, \
		DARKNET_CFG_FILE, DARKNET_WEIGHTS, KERAS, KERAS_LOCATION, SUBSCRIPTION_KEY, SHOW_RESPONSE, SHOW_IMAGES, \
		LABEL_NAME, ROTNET_LOCATION, ROTNET_MODEL_NAME, ROTNET_SAVE_FILE_NAME

args = parse_args()

# One of {DARKNET, KERAS} needs to be true
# Specifies which classifier to use
DARKNET = args.DARKNET
KERAS = args.KERAS

# One of {TESSERACT, COGNITIVE_SERVICES} needs to be true
# Specifies which OCR to use
TESSERACT = args.TESSERACT
COGNITIVE_SERVICES = args.COGNITIVE_SERVICES

############################################################################

##### Darknet Information - Change if necessary to fit your needs #####

if DARKNET:
	if args.DARKNET_BINARY_LOCATION == None:
		if os.name == 'nt':
			global popen_spawn
			from pexpect import popen_spawn
			DARKNET_BINARY_LOCATION = "darknet.exe"
		else:
			DARKNET_BINARY_LOCATION = "./darknet"
	else:
		DARKNET_BINARY_LOCATION = args.DARKNET_BINARY_LOCATION

	#### Change the following attributes if you move the files/weights ####
	DARKNET_THRESH    = args.DARKNET_THRESH
	DARKNET_DATA_FILE = args.DARKNET_DATA_FILE
	DARKNET_CFG_FILE  = args.DARKNET_CFG_FILE
	DARKNET_WEIGHTS   = args.DARKNET_WEIGHTS
	
#######################################################################

##### Keras Information - Change if necessary to fit your needs #####

elif KERAS:
	if os.name == 'nt':
		global popen_spawn
		from pexpect import popen_spawn

	# Change the location of Keras-yolo3 if you 
	# move it. You will need to change Keras-yolo's 
	# source code with the changes for the weights.
	KERAS_LOCATION = args.KERAS_LOCATION

#####################################################################

#### Cognitive Services Information ####

SUBSCRIPTION_KEY = args.SUBSCRIPTION_KEY
SHOW_RESPONSE = args.SHOW_RESPONSE

########################################

################ Locate_asset information ################

# Determines if we should show images after cropping them
SHOW_IMAGES = args.SHOW_IMAGES
# Name of the labels
LABEL_NAME = args.LABEL_NAME

##########################################################

########################## RotNet Constants ###########################
### The following constants will most likely not need to be changed ###

ROTNET_LOCATION			= args.ROTNET_LOCATION
ROTNET_MODEL_NAME		= args.ROTNET_MODEL_NAME
ROTNET_SAVE_FILE_NAME	= args.ROTNET_SAVE_FILE_NAME

#######################################################################

####################### DATABASE INFO #######################

LOCAL_DATABASE 	= args.LOCAL_DATABASE
COSMOS_DATABASE	= args.COSMOS_DATABASE

#############################################################

HOST = ""
MASTER_KEY = ""
DATABASE_ID = ""
COLLECTION_ID = ""