-
Notifications
You must be signed in to change notification settings - Fork 3
/
eval_raft_nc_sintel.sh
34 lines (33 loc) · 1.17 KB
/
eval_raft_nc_sintel.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
#!/bin/bash
#mkdir -p checkpoints
#if [ "$HOSTNAME" = "n2018" ] || [ "$HOSTNAME" = "n2017" ]; then
# rsync -avP /proj/cvl-compute-2020/users/x_abdel/datasets/kitti_flow.tar /scratch/local
# cd /scratch/local
# tar -xvf kitti_flow.tar
# cd /proj/cvl-compute-2020/users/x_abdel/code/RAFT
# ln -s /scratch/local/kitti_flow/data_scene_flow datasets/KITTI
#fi
python -u evaluate.py \
--model raft_nc_dbl \
--restore_ckpt models/raft_nc-sintel.pth \
--dataset sintel \
--final_upsampling=NConvUpsampler \
--final_upsampling_scale=4 \
--final_upsampling_use_data_for_guidance=True \
--final_upsampling_channels_to_batch=True \
--final_upsampling_use_residuals=False \
--final_upsampling_est_on_high_res=False \
--interp_net=NConvUNet \
--interp_net_channels_multiplier=2 \
--interp_net_num_downsampling=1 \
--interp_net_data_pooling="conf_based" \
--interp_net_encoder_filter_sz=5 \
--interp_net_decoder_filter_sz=3 \
--interp_net_out_filter_sz=1 \
--interp_net_shared_encoder=True \
--interp_net_use_double_conv=False \
--interp_net_use_bias=False \
--weights_est_net=Simple \
--weights_est_net_num_ch="[64, 32]" \
--weights_est_net_filter_sz="[3, 3, 1]" \
--weights_est_net_dilation="[1, 1, 1]" \