ashawkey/RAD-NeRF

How to generate .pth file?

xionghaoo opened this issue · 5 comments

./data/
├──.mp4 # original video
├──ori_imgs # original images from video
│ ├──0.jpg
│ ├──0.lms # 2D landmarks
│ ├──...
├──gt_imgs # ground truth images (static background)
│ ├──0.jpg
│ ├──...
├──parsing # semantic segmentation
│ ├──0.png
│ ├──...
├──torso_imgs # inpainted torso images
│ ├──0.png
│ ├──...
├──aud.wav # original audio
├──aud_eo.npy # audio features (wav2vec)
├──aud.npy # audio features (deepspeech)
├──bc.jpg # default background
├──track_params.pt # raw head tracking results
├──transforms_train.json # head poses (train split)
├──transforms_val.json # head poses (test split)

not found obama_eo.pth and obama.json file after training

me too

should be in trial_filename/checkpoints/
ex: trial_obama/checkpoints/

应该在 trial_filename/checkpoints/ 例如:trial_obama/checkpoints/

I couldn't find it. Can you provide a screenshot

I also couldn't find it using the notebook file explorer at first, but it was there when I did ls /notebooks/RAD-NeRF/trial_gary/checkpoints in the notebook.

can try scripts/train_obama_eo.sh