Wrapper to set Darknet parameters for Convolution2D.
(*args, **kwargs)
| 16 | |
| 17 | @wraps(Conv2D) |
| 18 | def DarknetConv2D(*args, **kwargs): |
| 19 | """Wrapper to set Darknet parameters for Convolution2D.""" |
| 20 | darknet_conv_kwargs = {'kernel_regularizer': l2(5e-4)} |
| 21 | darknet_conv_kwargs['padding'] = 'valid' if kwargs.get('strides')==(2,2) else 'same' |
| 22 | darknet_conv_kwargs.update(kwargs) |
| 23 | return Conv2D(*args, **darknet_conv_kwargs) |
| 24 | |
| 25 | def DarknetConv2D_BN_Leaky(*args, **kwargs): |
| 26 | """Darknet Convolution2D followed by BatchNormalization and LeakyReLU.""" |
no outgoing calls
no test coverage detected