Plugin？ Relu or PRelu

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3

MIT License

489 stars 165 forks source link

Plugin？ Relu or PRelu #73

Open zcc199710 opened 4 years ago

zcc199710 commented 4 years ago

The original version of yolo uses LRelu. Why does your plugin use PRelu, which will have a little impact on the accuracy of the results, and now many caffe-yolov3 use Relu

lewes6369 commented 4 years ago

The repo used LRelu, too.The comment in NvInferPlugin.h header which figures out the createPReLUPlugin function actually implemented as LeakyRelu not PRelu .I am not sure it is correct in recently version.