-
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
ValueError in confusion_matrix_tf2.py #35
Comments
Hey do you got solution?? Am facing same issue |
@harshit-777 No, I abandoned this codebase long ago. Try scripting your confusion matrix code using the bbox coors from the model outputs. |
I tried to run the confusion_matrix_tf2.py script to get CM on a trained model, but it runs into a ValueError. I also noticed the --inference_graph flag requires the path to the directory of saved_model.pb. This isn't clear from the documentation.
I tried executing:
!python confusion_matrix_tf2.py \ --input_tfrecord_path=/path/to/test.record \ --inference_graph=/path/to/directory/Models/saved_model \ --class_labels=/path/to/label_map.pbtxt \ --output_path=confusion_matrix.csv
After loading of the model and as evaluation starts, I get the following error:
The TFRecord file is created using the generate_tfrecord.py script from the official docs. The inference graph is created using exporter_main_V2.py.
Any help would be much appreciated since I need an urgent solution.
The text was updated successfully, but these errors were encountered: