Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,154 Bytes
7f51798 |
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 |
set -x
# n_proc=4
# n_points=4096
# focal=525
# pre-requisites: kaolin and nvdiffrast
# pip install kaolin==0.16.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.3.0_cu118.html
n_proc=1
n_points=4096
focal=24 # as in blender config
# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/02958343/fffb1660a38af30ba4cf3601fb6b2442/models/
# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/02958343/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd/car
# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/03001627/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd-full/chair
# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/02691156/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd/plane
# shape_root=/mnt/lustre/share/fzhong/shapenet/ShapeNetCore.v2/03001627/
# save_root=/mnt/lustre/yslan/3D_Dataset/shapenet/pcd-full/chair
# ! render gt
# shape_root=/mnt/sfs-common/yslan/Dataset/GSO/gso-unzip/gso-unzip
# save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/3D-metrics/gso
# ! use FPS later:
# splatter-img LGM
# for method_name in ln3diff One-2-3-45 OpenLRM shape-e Lara CRM splatter-img
# for method_name in ln3diff One-2-3-45 OpenLRM
# for method_name in shape-e Lara CRM splatter-img
# for method_name in CRM splatter-img
# for method_name in splatter-img LGM
# for method_name in splatter-img/Animals LGM/Animals
# for method_name in splatter-img LGM
for method_name in LGM_fixpose
do
# shape_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/${method_name}
# save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/3D-metrics/${method_name}
shape_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir-objv/${method_name}
save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir-objv/3D-metrics-fps/${method_name}
python fps_gs.py \
--shape_root ${shape_root} \
--save_root ${save_root} \
--n_proc ${n_proc} \
--n_points ${n_points} \
--image_height 512 \
--image_width 512 \
--focal_length_x ${focal} \
--focal_length_y ${focal} \
--principal_point_x 256 \
--principal_point_y 256 \
done |