|
def | __init__ (self, model, prev_blob, no_bias, is_test, bn_epsilon=1e-5, bn_momentum=0.9) |
|
def | add_conv (self, in_filters, out_filters, kernel, stride=1, group=1, pad=0) |
|
def | add_relu (self) |
|
def | add_spatial_bn (self, num_filters) |
|
def | add_bottleneck (self, input_filters, base_filters, output_filters, stride=1, group=1, spatial_batch_norm=True) |
|
def | add_simple_block (self, input_filters, num_filters, down_sampling=False, spatial_batch_norm=True) |
|
|
| model |
|
| comp_count |
|
| comp_idx |
|
| prev_blob |
|
| is_test |
|
| bn_epsilon |
|
| bn_momentum |
|
| no_bias |
|
Helper class for constructing residual blocks.
Definition at line 18 of file resnet.py.
The documentation for this class was generated from the following file: