Create aspp block
(high_level_ch, bottleneck_ch, output_stride, dpc=False)
| 299 | |
| 300 | |
| 301 | def get_aspp(high_level_ch, bottleneck_ch, output_stride, dpc=False): |
| 302 | """ |
| 303 | Create aspp block |
| 304 | """ |
| 305 | if dpc: |
| 306 | aspp = DPC(high_level_ch, bottleneck_ch, output_stride=output_stride) |
| 307 | else: |
| 308 | aspp = AtrousSpatialPyramidPoolingModule(high_level_ch, bottleneck_ch, |
| 309 | output_stride=output_stride) |
| 310 | aspp_out_ch = 5 * bottleneck_ch |
| 311 | return aspp, aspp_out_ch |
| 312 | |
| 313 | |
| 314 | def BNReLU(ch): |