[16dd74]: / dsb2018_topcoders / selim / predict_test.sh

Download this file

31 lines (27 with data), 1.1 kB

 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
#!/usr/bin/env bash
########## Resnet101 full masks for extension #############
python pred_test.py \
--gpu 0 \
--preprocessing_function caffe \
--network resnet101_2 \
--out_masks_folder pred_resnet101_full_masks \
--out_channels 2 \
--models_dir nn_models \
--models best_resnet101_2_fold0.h5 best_resnet101_2_fold1.h5 best_resnet101_2_fold2.h5 best_resnet101_2_fold3.h5
########## Densenet169 softmax 2 channels #############
python pred_test.py \
--gpu 0 \
--preprocessing_function torch \
--network densenet169_softmax \
--out_masks_folder pred_densenet169_softmax \
--out_channels 3 \
--models_dir nn_models \
--models best_densenet169_softmax_fold0.h5 best_densenet169_softmax_fold1.h5 best_densenet169_softmax_fold2.h5 best_densenet169_softmax_fold3.h5
########## Resnet152 sigmoid 2 channels #############
python pred_test.py \
--gpu 0 \
--preprocessing_function caffe \
--network resnet152_2 \
--out_masks_folder pred_resnet152 \
--out_channels 2 \
--models best_resnet152_2_fold0.h5 best_resnet152_2_fold1.h5 best_resnet152_2_fold2.h5 best_resnet152_2_fold3.h5