CAT-Seg / eval.sh
hsshin98
Add application file
d617811
raw
history blame
2.71 kB
#!/bin/sh
gpus=4
config=$1
output=$2
if [ -z $config ]
then
echo "No config file found! Run with "sh run.sh [CONFIG_FILE] [OUTPUT_DIR] [OPTS]""
exit 0
fi
if [ -z $output ]
then
echo "No output directory found! Run with "sh run.sh [CONFIG_FILE] [OUTPUT_DIR] [OPTS]""
exit 0
fi
shift 2
opts=${@}
#ADE20k-150
python train_net.py --config $config \
--num-gpus $gpus \
--dist-url "auto" \
--eval-only \
OUTPUT_DIR $output/eval \
MODEL.SEM_SEG_HEAD.TEST_CLASS_JSON "datasets/ADE_20k/ADE20K_150_class.json" \
DATASETS.TEST \(\"ade20k_150_test_sem_seg\"\,\) \
TEST.SLIDING_WINDOW "True" \
MODEL.SEM_SEG_HEAD.POOLING_SIZES "[1,1]" \
MODEL.WEIGHTS $output/model_final.pth \
$opts
#ADE20k-847
python train_net.py --config $config \
--num-gpus $gpus \
--dist-url "auto" \
--eval-only \
OUTPUT_DIR $output/eval \
MODEL.SEM_SEG_HEAD.TEST_CLASS_JSON "datasets/ADE_20k/ADE20K_847_pure_class.json" \
DATASETS.TEST \(\"ade20k_full_sem_seg_freq_val_all\"\,\) \
TEST.SLIDING_WINDOW "True" \
MODEL.SEM_SEG_HEAD.POOLING_SIZES "[1,1]" \
MODEL.WEIGHTS $output/model_final.pth \
$opts
#Pascal VOC
python train_net.py --config $config \
--num-gpus $gpus \
--dist-url "auto" \
--eval-only \
OUTPUT_DIR $output/eval \
MODEL.SEM_SEG_HEAD.TEST_CLASS_JSON "datasets/pascal-voc20/VOC_20_class.json" \
DATASETS.TEST \(\"voc_2012_test_sem_seg\"\,\) \
TEST.SLIDING_WINDOW "True" \
MODEL.SEM_SEG_HEAD.POOLING_SIZES "[1,1]" \
MODEL.WEIGHTS $output/model_final.pth \
$opts
#Pascal VOC-b
python train_net.py --config $config \
--num-gpus $gpus \
--dist-url "auto" \
--eval-only \
OUTPUT_DIR $output/eval \
MODEL.SEM_SEG_HEAD.TEST_CLASS_JSON "datasets/pascal-voc20/VOC_20_class_59.json" \
DATASETS.TEST \(\"voc_2012_test_openseg_sem_seg\"\,\) \
TEST.SLIDING_WINDOW "True" \
MODEL.SEM_SEG_HEAD.POOLING_SIZES "[1,1]" \
MODEL.WEIGHTS $output/model_final.pth \
$opts
#Pascal Context 59
python train_net.py --config $config \
--num-gpus $gpus \
--dist-url "auto" \
--eval-only \
OUTPUT_DIR $output/eval \
MODEL.SEM_SEG_HEAD.TEST_CLASS_JSON "datasets/pascal-context/pas59.json" \
DATASETS.TEST \(\"context_59_test_sem_seg\"\,\) \
TEST.SLIDING_WINDOW "True" \
MODEL.SEM_SEG_HEAD.POOLING_SIZES "[1,1]" \
MODEL.WEIGHTS $output/model_final.pth \
$opts
#Pascal Context 459
python train_net.py --config $config \
--num-gpus $gpus \
--dist-url "auto" \
--eval-only \
OUTPUT_DIR $output/eval \
MODEL.SEM_SEG_HEAD.TEST_CLASS_JSON "datasets/pascal-context/pas459.json" \
DATASETS.TEST \(\"context_459_test_sem_seg\"\,\) \
TEST.SLIDING_WINDOW "True" \
MODEL.SEM_SEG_HEAD.POOLING_SIZES "[1,1]" \
MODEL.WEIGHTS $output/model_final.pth \
$opts
cat $output/eval/log.txt | grep copypaste