zjumsj / FaceViewer

GNU General Public License v3.0
12 stars 1 forks source link

How to use pretrained model? #2

Open chiyic opened 1 week ago

chiyic commented 1 week ago

I just download the pretrain model of GaussianBlendshape, however, it includes many .npy and .pth files in zip file. So which one is useful for script: “--dataset=DATA_PATH/params --model=MODEL_PATH”? Is this checkpoint means the released pre-trained model in the script below? python scripts/extract_traj.py -i DATA_PATH/checkpoint -o DATA_PATH/params

zjumsj commented 1 week ago
  1. Use the parent directory of the *.pth file as MODEL_PATH
  2. No, there is a 'checkpoint' folder under any dataset directory. The script is used for converting the dataset.