(n, n_filter, filter_size=(3, 3), strides=(1, 1), name='conv_block')
| 23 | |
| 24 | |
| 25 | def conv_block(n, n_filter, filter_size=(3, 3), strides=(1, 1), name='conv_block'): |
| 26 | # ref: https://github.com/keras-team/keras/blob/master/keras/applications/mobilenet.py |
| 27 | n = Conv2d(n_filter, filter_size, strides, b_init=None, name=name + '.conv')(n) |
| 28 | n = BatchNorm(decay=0.99, act=tf.nn.relu6, name=name + '.batchnorm')(n) |
| 29 | return n |
| 30 | |
| 31 | |
| 32 | def depthwise_conv_block(n, n_filter, strides=(1, 1), name="depth_block"): |
no test coverage detected
searching dependent graphs…