File size: 1,995 Bytes
c5aa5a6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | ########################################################################################################################
# ENVIRONMENT VARIABLES #
########################################################################################################################
NUM_GPU = "NUM_GPU"
SAGEMAKER_INFERENCE_OUTPUT = "SAGEMAKER_INFERENCE_OUTPUT"
########################################################################################################################
# OUTPUT CONSTANTS #
########################################################################################################################
PROBABILITY = "probability"
PROBABILITIES = "probabilities"
PREDICTED_LABEL = "predicted_label"
LABELS = "labels"
########################################################################################################################
# DATA FORMAT CONSTANTS #
########################################################################################################################
BYTE_ARRAY_FORMAT = "application/x-image"
JPEG_FORMAT = "image/jpeg"
PNG_FORMAT = "image/png"
JSON_FORMAT = "application/json"
CSV_FORMAT = "text/csv"
COMMA_DELIMITER = ","
BRACKET_FORMATTER = '"{}"'
ALLOWED_INPUT_FORMATS = [BYTE_ARRAY_FORMAT, JPEG_FORMAT, PNG_FORMAT]
ALLOWED_OUTPUT_FORMATS = [JSON_FORMAT, CSV_FORMAT]
########################################################################################################################
# INFERENCE DATA CONSTANTS #
########################################################################################################################
IMAGE_COLUMN_NAME = "image"
|