search

qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
https://smp.readthedocs.io/
MIT License
9.58k stars 1.67k forks source link

UNetPlusPlus CenterBlock is not use in the current code #345

Closed Eric2Hamel closed 2 years ago

Eric2Hamel commented 3 years ago

I have checked your code for UNetPlusPlus and it seems that you do not use the centerBlock that you defined. I could not see any use in any ModuleDict or ModuleList or SequentialModel and neither in the forward method of the UNetPlusPlus decoder.

if center: self.center: nn.Module = CenterBlock(head_channels, head_channels, use_batchnorm=use_batchnorm, activation_name=activation_name) else: self.center = nn.Identity()

This center block is still valid? It seams to have center=True with vgg networks so it might do something wrong with vgg network currently.

github-actions[bot] commented 2 years ago

This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.

github-actions[bot] commented 2 years ago

This issue was closed because it has been stalled for 7 days with no activity.