Skip to content

激活函数改进 #242

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kz258852 opened this issue Feb 28, 2023 · 4 comments
Open

激活函数改进 #242

kz258852 opened this issue Feb 28, 2023 · 4 comments

Comments

@kz258852
Copy link

您好,我想将 yolo3/models/layers.py 里面的LeakyRelu激活函数替换成PRelu,正确的函数调用方式是什么呢,谢谢。

详细来说, 我想对这两个模块进行替换:
LeakyReLU(alpha=0.1, name='conv_dw_%s_leaky_relu' % block_id_str) ; LeakyReLU(alpha=0.1, name='conv_pw_%s_leaky_relu' % block_id_str)

PRelu基础用法:
tf.keras.layers.PReLU(
alpha_initializer='zeros', alpha_regularizer=None,
alpha_constraint=None, shared_axes=None, **kwargs
)

@david8862
Copy link
Owner

基本上应该可以直接替换,如

    return compose(
        DarknetDepthwiseConv2D(kernel_size, name='conv_dw_' + block_id_str, **no_bias_kwargs),
        CustomBatchNormalization(name='conv_dw_%s_bn' % block_id_str),
        LeakyReLU(alpha=0.1, name='conv_dw_%s_leaky_relu' % block_id_str),
        YoloConv2D(filters, (1,1), padding='same', use_bias=False, strides=(1, 1), name='conv_pw_%s' % block_id_str),
        CustomBatchNormalization(name='conv_pw_%s_bn' % block_id_str),
        LeakyReLU(alpha=0.1, name='conv_pw_%s_leaky_relu' % block_id_str))

替换为

    return compose(
        DarknetDepthwiseConv2D(kernel_size, name='conv_dw_' + block_id_str, **no_bias_kwargs),
        CustomBatchNormalization(name='conv_dw_%s_bn' % block_id_str),
        PReLU(name='conv_dw_%s_prelu' % block_id_str),
        YoloConv2D(filters, (1,1), padding='same', use_bias=False, strides=(1, 1), name='conv_pw_%s' % block_id_str),
        CustomBatchNormalization(name='conv_pw_%s_bn' % block_id_str),
        PReLU(name='conv_pw_%s_prelu' % block_id_str))

@kz258852
Copy link
Author

好的,我试一下,感谢您。我最近在用您的这个代码跑小目标检测,效果很好。我也想用您这个库中的部分代码,再加一点新的内容,也发一个GitHub,把自己对小目标检测的经验分享出来,您看可以吗
谢谢~

@david8862
Copy link
Owner

好的,我试一下,感谢您。我最近在用您的这个代码跑小目标检测,效果很好。我也想用您这个库中的部分代码,再加一点新的内容,也发一个GitHub,把自己对小目标检测的经验分享出来,您看可以吗 谢谢~

没问题。非常欢迎衍生出新工作:)

@kz258852
Copy link
Author

kz258852 commented Feb 28, 2023 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants