PolGS_plus/train.bash at main · PRIS-CV/PolGS_plus · GitHub

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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
output_dir="./exp/"

# PISR dataset
root_dir=''
list="StandingRabbit LyingRabbit"
for i in $list; do
    time=$(date "+%Y%m%d%H%M%S")
    python train.py \
        -s ${root_dir}${i} \
        -m ${output_dir}${i}_${time}  \
        --resolution 2 \
        --iterations 15000 \
        --stokes_l 0.1 \
        --azimuth_l 0.05 \
        --azimuth_start 1000\
    python render.py \
    -m ${output_dir}${i}_${time} \
    --resolution 2 \
    --img \
    --iteration -1
done

# SMVP3D dataset
root_dir=''
list="snail david hedgehog dragon squirrel"

for i in $list; do
    time=$(date "+%Y%m%d%H%M%S")
    python train.py \
        -s ${root_dir}${i} \
        -m ${output_dir}${i}_${time}  \
        --resolution 1 \
        --iterations 15000 \
        --stokes_l 0.1 \
        --azimuth_l 0.5 \
        --azimuth_start 1000
    python render.py \
    -m ${output_dir}${i}_${time} \
    --resolution 1 \
    --img \
    --iteration -1
done

# RMVP3D dataset
root_dir=''
list="shisa frog dog"
for i in $list; do
    time=$(date "+%Y%m%d%H%M%S")
    python train.py \
        -s ${root_dir}${i} \
        -m ${output_dir}${i}_${time}  \
        --resolution 2 \
        --iterations 15000 \
        --stokes_l 0.1\
        --azimuth_l 0.1 \
        --azimuth_start 1000\
    python render.py \
    -m ${output_dir}${i}_${time} \
    --resolution 2 \
    --img \
    --iteration -1
done

# PANDORA dataset
list="owl vase"
for i in $list; do
    time=$(date "+%Y%m%d%H%M%S")
    python train.py \
        -s ${root_dir}${i} \
        -m ${output_dir}${i}_${time}  \
        --resolution 4 \
        --iterations 15000 \
        --stokes_l 0.1 \
        --azimuth_l 0.1 \
        --azimuth_start 1000\

    python render.py \
    -m ${output_dir}${i}_${time} \
    --resolution 4 \
    --img \
    --iteration -1

done