Skip to content
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

mAP drops in the converted SSD mobilenet v2 #530

Open
dsnsabari opened this issue Jan 31, 2022 · 2 comments
Open

mAP drops in the converted SSD mobilenet v2 #530

dsnsabari opened this issue Jan 31, 2022 · 2 comments

Comments

@dsnsabari
Copy link

I followed all the steps, you have mentioned for converting TensorFlow SSD model v2 1.14 into TensorRT bin.
I updated the buildengine.py file and graphsurgeon converter for new Jetpack support.

Jetpack version is 4.6

Do you the reason for the wrong bounding box?

@dsnsabari dsnsabari reopened this Jan 31, 2022
@dsnsabari
Copy link
Author

dsnsabari commented Jan 31, 2022

@jkjung-avt , I trained the SSD mobilnet v2 model on tf 14.0 and tested the model on tf 15.4 with Jetpack version 4.6

I trained my custom model with Object 1 and Object 2. The Object 2accuracy was too low for tensorRT. Could you tell me, how to improve the tensorRT accuracy?

  Object 1 count Accuracy Object 2 Count Accuracy
TensorRT 41.48 94.6
TF frozen graph 86.807 93.2

@dsnsabari
Copy link
Author

As per your suggestion, I had removed the FP16 from the build_engine.py file. But still, I am getting the same accuracy. There is no improvement
#504 (comment)

@dsnsabari dsnsabari changed the title I am getting wrong bounding box in converted SSD mobilenet v2 mAP drops in the converted SSD mobilenet v2 Feb 1, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant