tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard

MAISI Unconditional UNET Pretrained Weights availability

Open QuantPrincess opened this issue 1 year ago • 3 comments

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?

QuantPrincess avatar Jan 31 '25 07:01 QuantPrincess

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

Can-Zhao avatar Mar 21 '25 05:03 Can-Zhao

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).

Masaaki-75 avatar Apr 07 '25 13:04 Masaaki-75

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.

Can-Zhao avatar Apr 09 '25 19:04 Can-Zhao