Tensorflow 모델을 Object Detection API에서 uff로 변환

Question

Mobilenet (및 TF OD API의 다른 모델)을 uff 형식으로 변환 한 다음 TensorRT 형식 인 because of much unsupported layers으로 변환 할 방법이 없다고합니다. 해당 레이어를 제거 \ 제거하는 방법이 있습니까? 예를 들어 graph_transform 도구를 사용하면 어쩌면? 나는이 모든 층의 목적이 아니라는 것을 이해합니다. 누군가가 시도하려는 경우 여기에 default model입니다.

Answer 1

UFF 변환 도구는 인식 할 수없는 일부 레이어에서 숨 막히는 것처럼 보입니다. 아마도 GA 릴리스에서 개선 될 것입니다. 지금은 그 레이어를 제거하고 추론에 필요한 가장 작은 하위 세트 만 유지 한 다음 nvinfer 플러그인 API를 사용하여 레이어를 구현해야합니다.