Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,040 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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
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
# n_points=35000 # will fps later
n_points=16384 # will fps later
# 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
# for method_name in gso
# for method_name in CRM splatter-img
# for method_name in GA/stage-2/dino_img/ditl-fromditlPCD-fixPose-tomesh GA/stage-2/dino_img/ditl-fromditlPCD-fixPose-tomesh-ditxlPCD
# '''
# for method_name in GA/stage-2/dino_img/ditl-fromditlPCD-fixPose-tomesh
# 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}
# save_root=/mnt/sfs-common/yslan/Repo/3dgen/FID-KID-Outputdir/3D-metrics-fps/${method_name}
# python sample_surface_gso.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
# '''
# for method_name in Lara CRM/Animals ln3diff-lite/Animals
# for method_name in Lara CRM/Animals ln3diff-lite/Animals
# for method_name in OpenLRM/Animals One-2-3-45/Animals shape-e/Animals
for method_name in scale3d/eval/eval_nerf/Animals scale3d/eval/eval_mesh/Animals
do
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 sample_surface_objv.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
|