search

open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
8.3k stars 2.62k forks source link

Make UNet backbone support dynamic shape #1236

Open RunningLeon opened 2 years ago

RunningLeon commented 2 years ago

Describe the feature

Motivation Currently, UNet backbone in mmseg could not be feed with different scale of images. Hope to support dynamic feature. Most related code is here https://github.com/open-mmlab/mmsegmentation/blob/e2361cbce04d90663752925cbbc92ee7ad7b331b/mmseg/models/utils/up_conv_block.py#L99

Related resources If there is an official code release or third-party implementations, please also provide the information here, which would be very helpful.

Additional context Add any other context or screenshots about the feature request here. If you would like to implement the feature and create a PR, please leave a comment here and that would be much appreciated.

MengzhangLI commented 2 years ago

OK, we would fix it as soon as possible.

MengzhangLI commented 2 years ago

Also need to modify:

https://github.com/open-mmlab/mmsegmentation/blob/e2361cbce04d90663752925cbbc92ee7ad7b331b/mmseg/ops/wrappers.py#L30