search

chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD， SSD（channel pruning）， Mobilenet-SSD
251 stars 84 forks source link

Are you planning to build the TensorRT-SSD with TensorFlow code? #19

Open foreverYoungGitHub opened 6 years ago

foreverYoungGitHub commented 6 years ago

Hi,

I just found that you put the TensorFlow wheel in the readme. Are you planning to build the TensorRT-SSD with TensorFlow code?

Best,

chenzhi1992 commented 6 years ago

Hi， I will do some work on accelerateing tensorflow model with TensorRT， but not Tensorflow SSD. I'm studying your MTCNN code， and it is very nice work.Thanks. @foreverYoungGitHub

zfc929 commented 6 years ago

tensorrt mtcnn process 1080p image ,average us timer 10ms,tensorrt ssd average 13ms

chenzhi1992 commented 6 years ago

wow, your work is very nice. Can you publish your code on your github and let‘s learn it?

foreverYoungGitHub commented 6 years ago

Thank you so much! Maybe my notification has some problem.

Your works are meaningful and provide a good example.

I'm also working on using tensorrt to accelerate the detection models. But currently, I'm using tensorflow and pytorch rather than caffe. And the MTCNN repo is not maintain for a long time...

Just let me know if you are interesting to working on model transformation between tensorflow/pytorch to tensorrt. Looking forward to cooperation with you.