MCPcopy Create free account
hub / github.com/CandleLabAI/PCBSegClassNet / bottleneck_block

Function bottleneck_block

src/models/blocks.py:119–147  ·  view source on GitHub ↗

bottleneck block for residual convolution Args: inputs: input to the bottleneck residual block filters: number of filters kernel_size: kernel size used in convolution temp: used in convolution loop: number of times residual bottleneck need to apply

(inputs,
                     filters=64,
                     kernel_size=(3, 3),
                     strides=(1, 1),
                     temp=6,
                     loop=3)

Source from the content-addressed store, hash-verified

117 return out
118
119def bottleneck_block(inputs,
120 filters=64,
121 kernel_size=(3, 3),
122 strides=(1, 1),
123 temp=6,
124 loop=3):
125 """
126 bottleneck block for residual convolution
127 Args:
128 inputs: input to the bottleneck residual block
129 filters: number of filters
130 kernel_size: kernel size used in convolution
131 temp: used in convolution
132 loop: number of times residual bottleneck need to apply
133 Returns:
134 out: output from residual bottleneck
135 """
136 out = residual_bottleneck(inputs=inputs,
137 filters=filters,
138 kernel_size=kernel_size,
139 strides=strides,
140 temp=temp)
141 for _ in range(1, loop):
142 out = residual_bottleneck(inputs=out,
143 filters=filters,
144 kernel_size=kernel_size,
145 temp=temp,
146 relu=True)
147 return out
148
149def pyramid_pooling_block(input_tensor, bin_sizes):
150 """

Callers 1

feature_extractorFunction · 0.85

Calls 1

residual_bottleneckFunction · 0.85

Tested by

no test coverage detected