set_env.sh

# 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