请问使用Relu layer, Scale layer 和 ElementWise替换prelu需要重新训练吗？

PKUZHOU / MTCNN_FaceDetection_TensorRT

MTCNN C++ implementation with NVIDIA TensorRT Inference accelerator SDK

202 stars 57 forks source link

Open guoguangchao opened 5 years ago

guoguangchao commented 5 years ago

请问使用Relu layer, Scale layer 和 ElementWise替换prelu需要重新训练吗？还是直接修改prototxt中对应的层就可以了？谢谢

PKUZHOU commented 5 years ago

不需要重新训练，但是需要写个脚本转换一下权重

guoguangchao commented 5 years ago

谢谢,转换权重是要加载caffemodel模型,将prelu层替换掉,再重新保存吗