TSMC/run.sh at main · SINRG-Lab/TSMC · 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
#!/bin/bash

export DOTNET_ROOT="$HOME/.dotnet"
export PATH="$HOME/.dotnet:$PATH"

DATASET="..."
NUM_FRAMES=10
NUM_CENTERS=2000

for GROUP_IDX in {1..10}; do
    echo "======================"
    echo " Running group $GROUP_IDX"
    echo "======================"

    cd tsmc/

    # --- Step 1: Get reference center ---
    python ./get_reference_center.py \
        --dataset $DATASET --num_frames $NUM_FRAMES --num_centers $NUM_CENTERS \
        --centers_dir ../arap-volume-tracking/data/combined-100-max-2000 \
        --group_idx $GROUP_IDX
       # --random_state 19056

    # --- Step 2: Get transformation ---
    python ./get_transformation.py \
        --dataset $DATASET --num_frames $NUM_FRAMES --num_centers $NUM_CENTERS \
        --centers_dir ../arap-volume-tracking/data/combined-100-max-2000 \
        --firstIndex 0 --lastIndex 9 \
        --group_idx $GROUP_IDX

    # --- Step 3: Run TVMEditor (stage 1) ---
    cd ..
    cd tvm-editing/
    TVMEditor.Test/bin/Release/net5.0/TVMEditor.Test $DATASET 1 0 9 \
        "./TVMEditor.Test/bin/Release/net5.0/Data/${DATASET}_2000/" \
        "./TVMEditor.Test/bin/Release/net5.0/output/${DATASET}_2000/"
    cd ..

    # --- Step 4: Extract reference mesh ---
    cd tsmc/
    python ./extract_reference_mesh.py \
        --dataset $DATASET --num_frames $NUM_FRAMES --num_centers $NUM_CENTERS \
        --inputDir ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/output/${DATASET}_2000/output/ \
        --outputDir ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/Data/${DATASET}_2000/reference_mesh/ \
        --firstIndex 0 --lastIndex 9 --key 4
    cd ..

    # --- Step 5: Run TVMEditor (stage 2) ---
    cd tvm-editing/
    TVMEditor.Test/bin/Release/net5.0/TVMEditor.Test $DATASET 2 0 9 \
        "./TVMEditor.Test/bin/Release/net5.0/Data/${DATASET}_2000" \
        "./TVMEditor.Test/bin/Release/net5.0/output/${DATASET}_2000"
    cd ..

    # --- Step 6: Displacements ---
    cd tsmc/
    python ./get_displacements.py \
        --dataset $DATASET --num_frames $NUM_FRAMES --num_centers $NUM_CENTERS \
        --target_mesh_path ../arap-volume-tracking/data/combined_scaled \
        --firstIndex 0 --lastIndex 9 \
        --group_idx $GROUP_IDX

    # --- Step 7: Compress displacements ---
    python compress_displacements.py \
        --dataset $DATASET --num_frames $NUM_FRAMES --num_eigenvectors 5 \
        --displacement_path ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/output/${DATASET}_2000/reference \
        --output_path ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/output/${DATASET}_2000/reference \
        --firstIndex 0 --lastIndex 9 \
        --reference_mesh_path ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/Data/${DATASET}_2000/reference_mesh/others/decoded_decimated_reference_mesh.obj

    # --- Step 8: Evaluation ---
    python evaluation.py \
        --dataset $DATASET --num_frames $NUM_FRAMES --num_centers $NUM_CENTERS \
        --input_path ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/output/${DATASET}_2000/reference \
        --dynamic_static_path ../data/$DATASET/meshes \
        --firstIndex 0 --lastIndex 9 \
        --reference_mesh_path ../tvm-editing/TVMEditor.Test/bin/Release/net5.0/Data/${DATASET}_2000/reference_mesh/others/decoded_decimated_reference_mesh.obj \
        --group_idx $GROUP_IDX

    cd ..

done