deepstream SSD

 

 

Pre-requisites: - Copy the model's label file "ssd_coco_labels.txt" from the data/ssd directory   in TensorRT samples to this directory. - Steps to generate the UFF model from ssd_inception_v2_coco TensorFlow frozen   graph. These steps have been referred from TensorRT sampleUffSSD README:   1. Make sure TensorRT's uff-converter-tf package is installed.   2. Install tensorflow-gpu package for python:      For dGPU:        $ pip install tensorflow-gpu      For Jetson, refer to https://elinux.org/Jetson_Zoo#TensorFlow   3. Download and untar the ssd_inception_v2_coco TensorFlow trained model from      http://download.tensorflow.org/models/object_detection/ssd_inception_v2_coco_2017_11_17.tar.gz   4. Navigate to the extracted directory and convert the frozen graph to uff:      $ cd ssd_inception_v2_coco_2017_11_17      $ python /usr/lib/python2.7/dist-packages/uff/bin/convert_to_uff.py \          frozen_inference_graph.pb -O NMS \          -p /usr/src/tensorrt/samples/sampleUffSSD/config.py \          -o sample_ssd_relu6.uff   5. Copy sample_ssd_relu6.uff to this directory.

$ python3 /usr/lib/python3.6/dist-packages/uff/bin/convert_to_uff.py

[링크 : https://docs.nvidia.com/metropolis/deepstream/dev-guide/text/DS_FAQ.html]

 

변환해서 netron.app 에서 보니 읭... output 이름이 NMS

타입과 텐서 차원이 안보인다?