-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy pathevaluate.sh
47 lines (39 loc) · 1.18 KB
/
evaluate.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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#!/bin/bash
# usage:
# ./evaluate.sh
echo "Evaluate..."
mode='evaluate'
# @200.121
DATA_DIR='/media/sdb1/chenrui/kitti_data/h5/'
# Where results (prediction plots and evaluation file) will be saved.
RESULTS_SAVE_DIR='./kitti_results/'
checkpoint_file='./checkpoint/checkpoint_epoch1_trLoss1342.3278.pkl' # load weights from checkpoint file for evaluating.
batch_size=10
num_plot=40 # how many images to plot.
# number of timesteps used for sequences in evaluating
num_timeSteps=10
workers=4
shuffle=false
data_format='channels_first'
n_channels=3
img_height=128
img_width=160
CUDA_VISIBLE_DEVICES=2 python evaluate.py \
--mode ${mode} \
--dataPath ${DATA_DIR} \
--resultsPath ${RESULTS_SAVE_DIR} \
--checkpoint_file ${checkpoint_file} \
--batch_size ${batch_size} \
--num_plot ${num_plot} \
--num_timeSteps ${num_timeSteps} \
--workers ${workers} \
--shuffle ${shuffle} \
--data_format ${data_format} \
--n_channels ${n_channels} \
--img_height ${img_height} \
--img_width ${img_width}
# --stack_sizes ${stack_sizes} \
# --R_stack_sizes ${R_stack_sizes} \
# --A_filter_sizes ${A_filter_sizes} \
# --Ahat_filter_sizes ${Ahat_filter_sizes} \
# --R_filter_sizes ${R_filter_sizes} \