feat(model): Add SuperADD VAND4.0 winning method#3628
Open
alexriedel1 wants to merge 5 commits into
Open
Conversation
This file contains hidden or 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
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.
📝 Description
Hello!
This adds the SuperADD as proposed in https://arxiv.org/abs/2605.14808 using the authors implementation https://github.com/LukasRoom/SuperADD
I made some minor changes to improve the memory consumption and runtime and to allow for batching during validation. Also the default image is set to 448 as 640 patches on the ViT H Super takes up extremely much memory.
I also removed all the "bells and whistles" like segmentation map post processing and training augmentations. The augmentations can easily be made with anomalibs datamodules and the segmentation map processing can be done manually if needed but only boost the score i guess.
With the default settings the model reaches:
MVTec AD2 Vial
MVTec AD2 Wallplug
✨ Changes
Select what type of change your PR is:
✅ Checklist
Before you submit your pull request, please make sure you have completed the following steps:
For more information about code review checklists, see the Code Review Checklist.