(n, n_filter, strides=(1, 1), name="depth_block")
| 30 | |
| 31 | |
| 32 | def depthwise_conv_block(n, n_filter, strides=(1, 1), name="depth_block"): |
| 33 | n = DepthwiseConv2d((3, 3), strides, b_init=None, name=name + '.depthwise')(n) |
| 34 | n = BatchNorm(decay=0.99, act=tf.nn.relu6, name=name + '.batchnorm1')(n) |
| 35 | n = Conv2d(n_filter, (1, 1), (1, 1), b_init=None, name=name + '.conv')(n) |
| 36 | n = BatchNorm(decay=0.99, act=tf.nn.relu6, name=name + '.batchnorm2')(n) |
| 37 | return n |
| 38 | |
| 39 | |
| 40 | def restore_params(network, path='models'): |
no test coverage detected
searching dependent graphs…