Skip to content

MAISI Unconditional UNET Pretrained Weights availability #1934

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.

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

Open
QuantPrincess opened this issue Jan 31, 2025 · 3 comments
Open

MAISI Unconditional UNET Pretrained Weights availability #1934

QuantPrincess opened this issue Jan 31, 2025 · 3 comments

Comments

@QuantPrincess
Copy link

I believe the weights provided for the unet in the inference tutorial are for unets that include control net. I'd like to do unconditional generation in a customised fashion and was wondering if the weights for a straight up unet (without control net) were available?

@Can-Zhao
Copy link
Contributor

Can-Zhao commented Mar 21, 2025

In https://github.com/Project-MONAI/tutorials/blob/main/generation/maisi/scripts/inference.py, the filename models/diff_unet_3d_*.pt are for unconditional generation, not for controlnet

@Masaaki-75
Copy link

Hi @Can-Zhao ! I am also interested in the pretrained model. I am wondering how to perform generation given only body region index without the body mask? If i understand it correctly, for now, the MAISI inference tutorial only shows mask-to-image inference (where the mask is either selected from the pre-built mask datasets or generated by another diffusion model).

@Can-Zhao
Copy link
Contributor

Can-Zhao commented Apr 9, 2025

In the end of https://github.com/Project-MONAI/tutorials/blob/main/generation/maisi/maisi_train_diff_unet_tutorial.ipynb , it includes scripts.diff_model_infer which can be used for unconditional generation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants