Skip to content
Snippets Groups Projects
calculate-class-weights.sh 417 B
Newer Older
#!/bin/bash

set -e

source prepare-shell.sh

echo "output will be written to: ${CLASS_WEIGHTS_URL}"

python -m sciencebeam_gym.tools.calculate_class_weights \
  --tfrecord-paths "${TRAIN_PREPROC_PATH}/*tfrecord*" \
  --image-key "annotation_image" \
  --color-map "${CONFIG_PATH}/${COLOR_MAP_FILENAME}" \
  --channels="$CHANNEL_NAMES" \
echo "output written to: ${CLASS_WEIGHTS_URL}"