[Feature] Support K-Net #1289
Merged
[Feature] Support K-Net #1289
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
4 tasks
|
After refactoring, the model can directly used for inference and result is the same, which means the key is still the same with old one.
|
Codecov Report
@@ Coverage Diff @@
## master #1289 +/- ##
==========================================
+ Coverage 90.32% 90.39% +0.06%
==========================================
Files 132 133 +1
Lines 7699 7879 +180
Branches 1290 1316 +26
==========================================
+ Hits 6954 7122 +168
- Misses 531 534 +3
- Partials 214 223 +9
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
|
Add ep3,4,5. |
|
Could anyone @ZwwWayne for code review? Thx in advance. |
ZwwWayne
reviewed
Mar 1, 2022
|
|
MeowZheng
reviewed
Mar 1, 2022
MeowZheng
reviewed
Mar 7, 2022
MeowZheng
reviewed
Mar 7, 2022
MeowZheng
reviewed
Mar 8, 2022
configs/knet/knet_s3_upernet_swin-l_8x2_512x512_adamw_80k_ade20k.py
Outdated
Show resolved
Hide resolved
configs/knet/knet_s3_upernet_swin-l_8x2_640x640_adamw_80k_ade20k.py
Outdated
Show resolved
Hide resolved
configs/knet/knet_s3_upernet_swin-t_8x2_512x512_adamw_80k_ade20k.py
Outdated
Show resolved
Hide resolved
Comment on lines
92
to
93
| Tensor: The output tensor of shape | ||
| (self.num_classes(KernelUpdateHead) * self.in_channels(KernelUpdateHead)/ in_channels, kernel size * kernel size, in_channels). # noqa |
There was a problem hiding this comment.
Tensor: The dynamic kernel with shape ()?
is there any short description for the shape of the dynamic kernel? @MengzhangLI @ZwwWayne
There was a problem hiding this comment.
Something like: 'The dynamic kernel with shape (N, C), where N is the number of classes, C is the feature map channels'?
MeowZheng
reviewed
Mar 8, 2022
MeowZheng
approved these changes
Mar 10, 2022
mob5566
pushed a commit
to mob5566/mmsegmentation
that referenced
this pull request
Apr 13, 2022
* knet first commit * fix import error in knet * remove kernel update head from decoder head * [Feature] Add kenerl updation for some decoder heads. * [Feature] Add kenerl updation for some decoder heads. * directly use forward_feature && modify other 3 decoder heads * remover kernel_update attr * delete unnecessary variables in forward function * delete kernel update function * delete kernel update function * delete kernel_generate_head * add unit test & comments in knet.py * add copyright to fix lint error * modify config names of knet * rename swin-l 640 * upload models&logs and refactor knet_head.py * modify docstrings and add some ut * add url, modify docstring and add loss ut * modify docstrings
ZhimingNJ
pushed a commit
to AetrexTechnology/mmsegmentation
that referenced
this pull request
Jun 29, 2022
* knet first commit * fix import error in knet * remove kernel update head from decoder head * [Feature] Add kenerl updation for some decoder heads. * [Feature] Add kenerl updation for some decoder heads. * directly use forward_feature && modify other 3 decoder heads * remover kernel_update attr * delete unnecessary variables in forward function * delete kernel update function * delete kernel update function * delete kernel_generate_head * add unit test & comments in knet.py * add copyright to fix lint error * modify config names of knet * rename swin-l 640 * upload models&logs and refactor knet_head.py * modify docstrings and add some ut * add url, modify docstring and add loss ut * modify docstrings
aravind-h-v
pushed a commit
to aravind-h-v/mmsegmentation
that referenced
this pull request
Mar 27, 2023
…pen-mmlab#1289) * fix non square images with UNet2DModel and DDIM/DDPM pipelines * fix unet_2d `sample_size` docstring * update pipeline tests for unet uncond Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Re-implementing K-Net.
Original paper: https://arxiv.org/abs/2106.14855.
Original repo:https://github.com/ZwwWayne/K-Net.
Results on ADE20K