-
Notifications
You must be signed in to change notification settings - Fork 25
/
Copy pathtest_modelnet40.sh
33 lines (27 loc) · 916 Bytes
/
test_modelnet40.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
#!/bin/bash
##########
# case 1 #
##########
# compute and save scores.
for c in `cat classes.txt`
do
python save_scores.py --center_only --gpu --model_def deploy_modelnet40_case1.prototxt --pretrained_model rotationnet_modelnet40_case1.caffemodel --input_file ModelNet40v1/test_$c.txt --output_file ModelNet40v1/test_$c.npy
done
# do prediction.
for c in `cat classes.txt`
do
python classify_npyfile_case1_all_views.py ModelNet40v1/test_$c.npy $c
done
##########
# case 2 #
##########
# compute and save scores.
for c in `cat classes.txt`
do
python save_scores.py --center_only --gpu --model_def deploy_modelnet40_case2.prototxt --pretrained_model rotationnet_modelnet40_case2.caffemodel --input_file ModelNet40v2/test_$c.txt --output_file ModelNet40v2/test_$c.npy
done
# do prediction.
for c in `cat classes.txt`
do
python classify_npyfile_case2_all_views.py ModelNet40v2/test_$c.npy $c
done