I have re-trained the "DetectNet model with my dataset using DIGITS.
However, when I try to use that model with the “object detection” sample provided with Driveworks, it is giving me an error:
Setting up bbox/regressor
Top shape: 1 4 24 78 (7488)
Memory required for data: 636704640
Creating layer cluster
Unknown layer type: Python (known types: AbsVal, ArgMax, BNLL, BatchNorm, BatchReindex, Bias, Concat, Convolution, Crop, Deconvolution, Dropout, ELU, Eltwise, Embed, Exp, Filter, Flatten, Im2col, InnerProduct, Input, LRN, Log, MVN, PReLU, Pooling, Power, ReLU, Reduction, Reshape, SPP, Scale, Sigmoid, Silence, Slice, Softmax, Split, TanH, Threshold, Tile)
Segmentation fault (core dumped)
I had already compiled my caffe with the PYTHON_LAYER flag set to true before training detectnet.
I am still getting this error with driveworks sample.
Please suggest any workaround.