# environment variables for model training | |
# environment variables for the inference api | |
export DATA_DIR=queries | |
export PATCHES_DIR=${DATA_DIR}/patches | |
export SLIDES_DIR=${DATA_DIR}/slides | |
export GRAPHCAM_DIR=${DATA_DIR}/graphcam_plots | |
mkdir $GRAPHCAM_DIR -p | |
# manually put the metadata in the metadata folder | |
export CLASS_METADATA='metadata/label_map.pkl' | |
# manually put the desired weights in the weights folder | |
export WEIGHTS_PATH='weights' | |
export FEATURE_EXTRACTOR_WEIGHT_PATH=${WEIGHTS_PATH}/feature_extractor/model.pth | |
export GT_WEIGHT_PATH=${WEIGHTS_PATH}/graph_transformer/GraphCAM.pth |