search

AITTSMD / MTCNN-Tensorflow

Reproduce MTCNN using Tensorflow
1.51k stars 711 forks source link

How to convert the three models of PNet, RNet, Onet into single pb file? #268

Open BrownOfSummer opened 5 years ago

BrownOfSummer commented 5 years ago

How to convert the three models of PNet, RNet, Onet into single pb file, so it simple to deploy in C++?

xiaqing10 commented 5 years ago

Using three models can also deployed simply in C++.

chenkeyi2013 commented 5 years ago

How to convert the three models of PNet, RNet, Onet into single pb file, so it simple to deploy in C++?

I think it will be little complex to convert three model to one pb file. A pb file looks like only contain tensor, tensor'op and their paramter.

MTCNN needs run NMS after run PNet and RNet. But in this project looks like NMS code with numpy,not tensorflow, the PRO Net connect part code with python as well..

therefore,maybe you need define your own tensor NMS and use tensor connect three model.

heisherhub commented 4 years ago

Using three models can also deployed simply in C++.

How to convert the three models of PNet, RNet, Onet into single pb file, so it simple to deploy in C++?

I think it will be little complex to convert three model to one pb file. A pb file looks like only contain tensor, tensor'op and their paramter.

MTCNN needs run NMS after run PNet and RNet. But in this project looks like NMS code with numpy,not tensorflow, the PRO Net connect part code with python as well..

therefore,maybe you need define your own tensor NMS and use tensor connect three model.

Tensorflow can do NMS, the difficult is how to do image pyramid