| ######################################################################################################################## | |
| # 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" | |