search

Xharlie / DISN

(latest updates and bug fixed) DISN: Deep Implicit Surface Network for High-quality Single-view 3D Reconstruction
183 stars 27 forks source link

ValueError: Invalid checkpoint state loaded from #16

Closed JohnG0024 closed 3 years ago

JohnG0024 commented 3 years ago

Hi, if I use new rendered 2d image dataset, I should just go through the last two bulletpoint commands in section Camera parameters estimation network right?

But I'm getting 

Traceback (most recent call last):
  File "cam_est/train_sdf_cam.py", line 814, in <module>
    train()
  File "cam_est/train_sdf_cam.py", line 483, in train
    ckptstate = tf.train.get_checkpoint_state(PRETRAINED_MODEL_PATH)
  File "/home/johnG/anaconda3/envs/tf_1.13.0/lib/python3.6/site-packages/tensorflow/python/training/checkpoint_management.py", line 278, in get_checkpoint_state
    + checkpoint_dir)
ValueError: Invalid checkpoint state loaded from

when running this command

 python -u cam_est/train_sdf_cam.py --log_dir cam_est/checkpoint/{your checkpoint dir} --gpu 0 --loss_mode 3D --learning_rate 1e-4 --src_h5_dir {your new rendered images' h5 directory} --img_h 224 --img_w 224

Any idea how to solve this?