forked from XuelianCheng/LEAStereo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
predict_sf.sh
executable file
·15 lines (14 loc) · 987 Bytes
/
predict_sf.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
CUDA_VISIBLE_DEVICES=1 python predict.py \
--sceneflow=1 --maxdisp=192 \
--crop_height=576 --crop_width=960 \
--data_path='./dataset/SceneFlow/' \
--test_list='./dataloaders/lists/sceneflow_test.list' \
--save_path='./predict/sceneflow/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/sceneflow/best/checkpoint/best.pth'