Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Make UNet backbone support dynamic shape #1236

Open
RunningLeon opened this issue Jan 25, 2022 · 2 comments
Open

Make UNet backbone support dynamic shape #1236

RunningLeon opened this issue Jan 25, 2022 · 2 comments
Assignees
@MengzhangLI
Labels
enhancement New feature or request WIP Work in process

Comments

@RunningLeon
Copy link
Collaborator

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

mmsegmentation/mmseg/models/utils/up_conv_block.py

Line 99 in e2361cb

out = torch.cat([skip, x], dim=1)

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
Copy link
Contributor

OK, we would fix it as soon as possible.

@MengzhangLI MengzhangLI added WIP Work in process enhancement New feature or request labels Jan 25, 2022
@MengzhangLI MengzhangLI self-assigned this Jan 25, 2022
@RunningLeon RunningLeon mentioned this issue Jan 25, 2022
Merged
3 tasks
@MengzhangLI
Copy link
Contributor

Also need to modify:

mmsegmentation/mmseg/ops/wrappers.py

Line 30 in e2361cb

class Upsample(nn.Module):

wjkim81 pushed a commit to wjkim81/mmsegmentation that referenced this issue Dec 3, 2023
@bladrome @ly015
[Fix] complete training hooks. (open-mmlab#1236)
663cfc8 
* Fix: complete training hooks.

* not expose timer_hook

Co-authored-by: ly015 <liyining0712@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@MengzhangLI MengzhangLI

Labels
enhancement New feature or request WIP Work in process
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@MengzhangLI @RunningLeon