GeorgeSeif / Semantic-Segmentation-Suite

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!
2.51k stars 880 forks source link

Checkpoint and model name responding relationship #139

Closed LittleByLittle2 closed 5 years ago

LittleByLittle2 commented 5 years ago

Please fill out this issue template before submitting. Issues which do not fill out this template, or are already answered in the FAQs will simply be closed.

Please go to Stack Overflow for help and support. Also check past issues as many are repeats. Also check out the Frequently Asked Questions (FAQs) below in case your question has already been answered in an issue!

Issues should be one of the following:

  1. Feature requests
  2. Bug reports

Information

Please specify the following information when submitting an issue:

Describe the problem

Describe the problem clearly here. Be sure to convey here why it's a bug or a feature request.

Some tensor name from the module is not in checkpoint. So is there any responding relationship between checkpoint and model name ?

Source code / logs

Include any logs or source code that would be helpful to diagnose the problem. If including tracebacks, please include the full traceback. Large logs and files should be attached. Try to provide a reproducible test case that is the bare minimum necessary to generate the problem.

NotFoundError (see above for traceback): Tensor name "inception_v4/Mixed_3a/Branch_1/Conv2d_0a_3x3/BatchNorm/beta" not found in checkpoint files models/resnet_v2_101.ckpt [[Node: save/RestoreV2_24 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2_24/tensor_names, save/RestoreV2_24/shape_and_slices)]] [[Node: save/RestoreV2_560/_121 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name="edge_1344_save/RestoreV2_560", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]]

FAQs

GeorgeSeif commented 5 years ago

The checkpoint path you have provided is for a classification network used for feature extraction. If you want to run prediction for segmentation, you'll need to train a model first! :)