forked from gustavz/realtime_object_detection
-
Notifications
You must be signed in to change notification settings - Fork 36
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Error with the splitting of the Faster RCNN Inception-V2 #76
Comments
By using Tensorboard, I saw that the correct names are probably: But I then got another error:
However, I clearly see that it seem correct in the original graph with Tensorboard, so I don't understand how to solve this error. The code also work well without the splitting. |
PoissonChasseur
changed the title
split_model for Faster RCNN Inception-V2 not working due to wrong keys in SPLIT_TARGET_NAME
Error with the splitting of the Faster RCNN Inception-V2
May 10, 2022
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hello,
I'm trying to use your code to split into 2 a "Faster RCNN Inception-V2" that I trained with the standard code from the "Object Detection API of Tensorflow" with Tensorflow 1.12.0.
When I try to use your code with my "frozen_inference_graph.pb" file, I have an error in the "load_graph_faster_v2.py" file because of the "Squeeze_2" and "Squeeze_3" keys (defined in "SPLIT_TARGET_NAME") which are not present in the "name_to_node_map" variable.
I looked at the contents of the "name_to_node_map" variable (that I put in a text file), but there are several squeeze layers that could relate to which one should be selected, so I'm not sure which ones should be selected in order to do it correctly.
Note: I'm a newbie to Tensorflow and object detection models.
The text was updated successfully, but these errors were encountered: