RAD-NeRF icon indicating copy to clipboard operation
RAD-NeRF copied to clipboard

How to generate .pth file?

Open xionghaoo opened this issue 1 year ago • 5 comments

./data/<ID> ├──<ID>.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

xionghaoo avatar Apr 26 '23 08:04 xionghaoo

me too

sunmingyang666 avatar Apr 27 '23 14:04 sunmingyang666

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

davidspicy avatar May 14 '23 21:05 davidspicy

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

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

weihaigang avatar Jun 03 '23 02:06 weihaigang

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.

Marvinified avatar Jun 13 '23 22:06 Marvinified

can try scripts/train_obama_eo.sh

xionghaoo avatar Jun 16 '23 17:06 xionghaoo