-
Notifications
You must be signed in to change notification settings - Fork 51
/
predict_md.sh
executable file
·15 lines (14 loc) · 986 Bytes
/
predict_md.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
CUDA_VISIBLE_DEVICES=0 python predict.py \
--middlebury=1 --maxdisp=408 \
--crop_height=1008 --crop_width=1512 \
--data_path='./dataset/MiddEval3/testH/' \
--test_list='./dataloaders/lists/middeval3_test.list' \
--save_path='./predict/middlebury/images/' \
--fea_num_layer 6 --mat_num_layers 12\
--fea_filter_multiplier 8 --fea_block_multiplier 4 --fea_step 3 \
--mat_filter_multiplier 8 --mat_block_multiplier 4 --mat_step 3 \
--net_arch_fea='run/sceneflow/best/architecture/feature_network_path.npy' \
--cell_arch_fea='run/sceneflow/best/architecture/feature_genotype.npy' \
--net_arch_mat='run/sceneflow/best/architecture/matching_network_path.npy' \
--cell_arch_mat='run/sceneflow/best/architecture/matching_genotype.npy' \
--resume './run/MiddEval3/best/best.pth'