Port explicit denoising methods from fmri-fm.archive#7
Open
har5h1l wants to merge 4 commits intoMedARC-AI:mainfrom
Open
Port explicit denoising methods from fmri-fm.archive#7har5h1l wants to merge 4 commits intoMedARC-AI:mainfrom
har5h1l wants to merge 4 commits intoMedARC-AI:mainfrom
Conversation
- Add denoising_demo.ipynb with updated imports and model loading - Fix patch_embed references in models_mae.py to use encoder.patchify - Support both new pretrained models and fallback to scratch initialization - Update notebook to work with new flat_mae module structure
…the denoising_demo notebook to be compatible.
- Implement dynamic module reloading in denoising_demo.ipynb for updated model imports. - Add pixel_mask creation for visualization in denoising_demo.ipynb. - Update print statements to include detailed output shapes and processing times. - Modify models_mae.py to ensure compatibility with input data types and improve patch handling. - Refactor patch_mask creation to accommodate both 2D and 3D cases, ensuring dtype consistency.
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
1 participant
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.
This PR ports the work on the explicit denoising methods in models_mae.py and its demo notebook from fmri-fm.archive.
denoising_demo.ipynb currently doesn't work, as it's still in the process of being updated for the newer repo codebase.
This is not intended to be pushed to main, there should be a new branch called dev/denoising made when merging.