Open alex88o opened 5 years ago
can you share with us your error msg ?
Loading model checkpoint weights ...
Traceback (most recent call last):
File "test.py", line 52, in
Caused by op u'save_1/RestoreV2_6', defined at:
File "test.py", line 51, in
NotFoundError (see above for traceback): Tensor name "BatchNorm_2/beta" not found in checkpoint files ./utils/models/resnet_v2_101.ckpt [[Node: save_1/RestoreV2_6 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/cpu:0"](_arg_save_1/Const_0_0, save_1/RestoreV2_6/tensor_names, save_1/RestoreV2_6/shape_and_slices)]]
"resnet_v2_101.ckpt" seems a pre-trained network that is loaded when building the network.
"--checkpoint_path" is "The path to the latest checkpoint weights for your model". That means the checkpoint of a supported trained network listed on main page...
I guess you have to delete the older checkpoints folder and re-train your model.
I didn't trained any model. I just want to download a pre-trained model and test it.
There is no pre-trained model provided in this project. You have to train the model by yourself.
Maybe by "download the pretrined model" he wanted to say the checkpoints (of the front ends)
Unfortunatily there is no support in this repo :(
can you share me all the weights with get_pretrained_checkpoints ? I don't know why i can't download these model
I downloaded all the weights with get_pretrained_checkpoints and they are all stored in /utils/models. Then, trying to run test.py with ANY combination of model/checkpoints it raises an error when loading the model, saying that some Tensor is not present in the weight file. An example of run command is:
python test.py --checkpoint_path ./utils/models/resnet_v2_101.ckpt --model custom