I am trying to get the tensorflow Resnet50 object detection model working with deepstream. I have tried to get the objectDetector_SSD example working with a Resnet50 model. I am working from the
First I verified the example as specified in the README works.
Then I attempted to convert the
ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync_2018_07_03 model to uff, which failed. I modified the config as described by liuyoungshop here:
Conversion to uff was successful after modifying the config.
However, when I try to run the gstreamer pipeline I get an error:
0:00:02.111132286 1764 0x55e82b8b80c0 ERROR UffParser: Parser error: FeatureExtractor/resnet_v1_50/fpn/top_down/nearest_neighbor_upsampling/mul: Invalid scale mode, nbWeights: 4 > 0:00:02.117444937 1764 0x55e82b8b80c0 ERROR nvinfer gs tnvinfer.cpp:511:gst_nvinfer_logger:<nvinfer0> NvDsInferContext[UID 1]:generateTRTModel(): Failed to parse UFF file: incorrect file or incorrect input/output blob names
I suppose this is because nearest neighbor resize isn’t supported?
Looks like tensorflow ResizeNearest op is supported in the latest tensorrt (6.01)
My questions are:
- Has anyone had success getting other ssd object detection models to convert to uff, especially resnet50? What did you do to get it to work?
- Has anyone had success writing plugins that use tftrt? Is there any planned support for tftrt in deepstream?
- Is there a timeline on a docker image becoming available with tensorrt 6.01 + deepstream?