Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Convert Tensorflow models from Object Detection API to uff

Seems, that there is no way to convert from the box Mobilenet (and other models from TF OD API) to uff format and then to TensorRT format, because of much unsupported layers. Is there some way to remove \ replace that layers? For example with graph_transform tool maybe? I understand the purpose of not all these layers. Here is the default model, if someone wants to try.

like image 781
UndeadDragon Avatar asked Oct 29 '22 23:10

UndeadDragon


1 Answers

The UFF conversion tools seems to choke on some unrecognized layers, perhaps this will be improved in the GA release. For now, you will need to remove those layers (And keep only the smallest subset needed for inference), then implement those using the nvinfer plugin API.

like image 54
tdeegan Avatar answered Jan 02 '23 21:01

tdeegan