|
|
|
|
|
|
|
|
TRAIN_CONFIG="configs/inference/lam-20k-8gpu.yaml" |
|
|
MODEL_NAME="model_zoo/lam_models/releases/lam/lam-20k/step_045500/" |
|
|
IMAGE_INPUT="assets/sample_input/status.png" |
|
|
MOTION_SEQS_DIR="assets/sample_motion/export/Look_In_My_Eyes/" |
|
|
|
|
|
|
|
|
TRAIN_CONFIG=${1:-$TRAIN_CONFIG} |
|
|
MODEL_NAME=${2:-$MODEL_NAME} |
|
|
IMAGE_INPUT=${3:-$IMAGE_INPUT} |
|
|
MOTION_SEQS_DIR=${4:-$MOTION_SEQS_DIR} |
|
|
|
|
|
echo "TRAIN_CONFIG: $TRAIN_CONFIG" |
|
|
echo "IMAGE_INPUT: $IMAGE_INPUT" |
|
|
echo "MODEL_NAME: $MODEL_NAME" |
|
|
echo "MOTION_SEQS_DIR: $MOTION_SEQS_DIR" |
|
|
|
|
|
|
|
|
MOTION_IMG_DIR=null |
|
|
SAVE_PLY=false |
|
|
SAVE_IMG=false |
|
|
VIS_MOTION=false |
|
|
MOTION_IMG_NEED_MASK=true |
|
|
RENDER_FPS=30 |
|
|
MOTION_VIDEO_READ_FPS=30 |
|
|
EXPORT_VIDEO=true |
|
|
CROSS_ID=false |
|
|
TEST_SAMPLE=false |
|
|
GAGA_TRACK_TYPE="" |
|
|
|
|
|
device=0 |
|
|
nodes=0 |
|
|
|
|
|
export PYTHONPATH=$PYTHONPATH:$pwd |
|
|
|
|
|
|
|
|
CUDA_VISIBLE_DEVICES=$device python -m lam.launch infer.lam --config $TRAIN_CONFIG \ |
|
|
model_name=$MODEL_NAME image_input=$IMAGE_INPUT \ |
|
|
export_video=$EXPORT_VIDEO export_mesh=$EXPORT_MESH \ |
|
|
motion_seqs_dir=$MOTION_SEQS_DIR motion_img_dir=$MOTION_IMG_DIR \ |
|
|
vis_motion=$VIS_MOTION motion_img_need_mask=$MOTION_IMG_NEED_MASK \ |
|
|
render_fps=$RENDER_FPS motion_video_read_fps=$MOTION_VIDEO_READ_FPS \ |
|
|
save_ply=$SAVE_PLY save_img=$SAVE_IMG \ |
|
|
gaga_track_type=$GAGA_TRACK_TYPE cross_id=$CROSS_ID \ |
|
|
test_sample=$TEST_SAMPLE rank=$device nodes=$nodes |
|
|
|
