tensorRT_optimization does not parse Slice Caffe layers

Autonomous Vehicles DRIVE PX 2 General
1

I’m trying to run driveworks-0.3/tools/sample_object_tracker on a custom .prototxt
and .caffemodel.

However despite what the documentation at driveworks/doc/nvdwx_html/dwx_object_tracker_sample.html indicates, --caffemodel
and --prototxt options are not outputted by ./sample_object_tracker --help.

Furthermore, when trying to convert my .caffemodel
to a tensorRT_model with ./tensorRT_optimization I receive the error

could not parse layer Slice

There is documentation for this layer at Caffe | Slice Layer but
I’m supposing that this layer isn’t supported by Driveworks currently?

2

Hello droberts42

Could you please try to run the sample with below command?
$ ./ sample_object_tracker --caffe_prototxt= --caffe_model=

$ ./tensorRT_optimization --prototxt= --caffemodel= --outputLayers=layer name(like coverage,bboxes)

Thanks.

3

Hi SteveNV, thanks for your reply.

When running

$ ./sample_object_tracker --caffe_prototxt=<path to my caffe prototxt file> --caffe_model=<path to my caffemodel file>

I get exactly the same terminal and video output as

./sample_object_tracker

When I run

./sample_object_tracker --help

I get

Options:
--tensorRT_model=../data/samples/detector/tensorRT_model.bin
--tracker=../data/samples/tracker/config.txt
--video=../data/samples/sfm/triangulation/video_0.h264
--yuv2rgb=cuda

There does not seem to be any option associated with .caffemodel or .prototxt input, as I stated in my previous post.

When I run

./tensorRT_optimization --prototxt=<path to my caffe prototxt file> --caffemodel=<path to my caffemodel file> --outputLayers=bboxes

I get

ProgramArguments: Missing required arguments: "outputBlobs",  e.g.
	../tools/tensorRT_optimization  --outputBlobs=<value>

So I tried

./tensorRT_optimization --prototxt=<path to my caffe prototxt file> --caffemodel=<path to my caffemodel file> --outputBlobs=bboxes

and I get

Initializing network optimizer on model <path to my caffe prototxt file> with weights from <path to my caffemodel file>
could not parse layer type Slice
Error: Provided network could not be parsed.

as I stated in my previous post.

4

Hello droberts42,

Could you please try to re-run after delete “Slice” in your caffe prototxt file?
I think it seems “Slice” has some different in the file. Thanks.

5

I have the same problem like yours.
Have you find the solution yet?

6

Dear lwy8976,

Did you test with the latest DPX2 PDK version?
Could you please let us know more detail info like PDK version, using command, error log, etc? Thanks.