-
Notifications
You must be signed in to change notification settings - Fork 9
/
02_run_acregnet.sh
executable file
·28 lines (24 loc) · 932 Bytes
/
02_run_acregnet.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/bin/bash
DATASET='JSRT'
python scripts/acregnet/train.py \
--train_images_file=data/$DATASET/split/train_images_256.txt \
--train_labels_file=data/$DATASET/split/train_labels_256.txt \
--autoencoder_file=results/$DATASET/AENet/train/model.pt \
--epochs=10 \
--steps_per_epoch=100 \
--batch_size=32 \
--lr=1e-3 \
--flow_weight=5e-5 \
--label_weight=1.0 \
--shape_weight=1e-1 \
--results_dir=results/$DATASET/ACRegNet/train \
--save_model
python scripts/acregnet/test.py \
--test_images_file=data/$DATASET/split/test_images_256.txt \
--test_labels_file=data/$DATASET/split/test_labels_256.txt \
--model_file=results/$DATASET/ACRegNet/train/model.pt \
--results_dir=results/$DATASET/ACRegNet/test \
--save_images
python scripts/acregnet/compute_metrics.py \
--results_dir=results/$DATASET/ACRegNet/test \
--output_dir=results/$DATASET/ACRegNet/metrics