create engine failed by my mobilenet_yolov3_lite caffemodel

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3

MIT License

489 stars 165 forks source link

create engine failed by my mobilenet_yolov3_lite caffemodel #53

Closed guods closed 5 years ago

guods commented 5 years ago

I train the mobilenet_yolov3_lite model by https://github.com/eric612/Caffe-YOLOv3-Windows, but it failed when I create engine file, does not parser "yolov3_detection_output_param". Can you help me, thanks.

lewes6369 commented 5 years ago

Change the last layer to type Yolo as readme described