AioMedica2 / set_env.sh
chris1nexus
First commit
54660f7
# 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