PengfeiMa

Hi, @maudzung I have tried to use the TRT to speed up the inference, but there is some trouble in the 'yolo_layer'. When I export ONNX model with yolo_layer, the...

add the DORN caffe python path to your ~/.bashrc `vim ~/.bashrc` add `export PYTHONPATH=="/your dir/DORN/caffe/python:/your dir/DORN/caffe/pylayer" ` `source ~/.bashrc`

try `opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.bool()` ----> `opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.byte()`

@rolan123 What's your final results on per class?

@rolan123 I test my epoch-70 and epoch-80, the results are almost the same as yours.

tf1.14 is ok, meet the same problem, just do some changes in the code. you can search the key words ' collection ' in Issues to get the way to...