search

alex04072000 / ObstructionRemoval

[CVPR 2020] Learning to See Through Obstructions
1k stars 187 forks source link

ModuleNotFoundError: No module named 'ckpt_mgr' #2

Closed aa12356jm closed 4 years ago

aa12356jm commented 4 years ago

Traceback (most recent call last): File "train_reflection_online.py", line 45, in <module> from model_pwcnet import ModelPWCNet, _DEFAULT_PWCNET_TEST_OPTIONS File "tfoptflow/tfoptflow/model_pwcnet.py", line 18, in <module> from model_base import ModelBase File "tfoptflow/tfoptflow/model_base.py", line 16, in <module> from ckpt_mgr import BestCheckpointSaver ModuleNotFoundError: No module named 'ckpt_mgr'

alex04072000 commented 4 years ago

You need to git clone the PWC-Net and pre-trained weight and overwrite tfoptflow/model_pwcnet.py and tfoptflow/model_base.py in this repository. Please refer to a working colab for reference.

aa12356jm commented 4 years ago

thanks

k4rth33k commented 4 years ago

The colab gives this error while running the fence_demo file. The model doesn't seem to be present in the needed folder. ValueError: The passed save_path is not a valid checkpoint: tfoptflow/tfoptflow/models/pwcnet-lg-6-2-multisteps-chairsthingsmix/pwcnet.ckpt-595000

reyking commented 3 years ago

The colab gives this error while running the fence_demo file. The model doesn't seem to be present in the needed folder. ValueError: The passed save_path is not a valid checkpoint: **_tfoptflow/tfoptflow/models/pwcnet-lg-6-2-multisteps-chairsthingsmix/_**pwcnet.ckpt-595000

go inside that folder there is a link to de model

fpjr commented 2 years ago

The wget cell must be placed before the python cell. Moreover, the command should be !wget not $ wget, and the output path tfoptflow/tfoptflow/models/pwcnet-lg-6-2-multisteps-chairsthingsmix/pwcnet.ckpt-595000.index