-
Notifications
You must be signed in to change notification settings - Fork 70
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
Porting of the X-Ray Latent Diffusion Models model from MONAI Generative Models into model-zoo. #718
base: dev
Are you sure you want to change the base?
Porting of the X-Ray Latent Diffusion Models model from MONAI Generative Models into model-zoo. #718
Conversation
…s repo. Signed-off-by: Virginia <[email protected]>
… that no longer applies.
for more information, see https://pre-commit.ci
models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json
Outdated
Show resolved
Hide resolved
models/cxr_image_synthesis_latent_diffusion_model/configs/metadata.json
Outdated
Show resolved
Hide resolved
models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json
Outdated
Show resolved
Hide resolved
models/cxr_image_synthesis_latent_diffusion_model/docs/README.md
Outdated
Show resolved
Hide resolved
… longer necessary. On inference, modification of the entry point from save_jpg to run to allow for run to be called alone, and truncation of loading statements which are too long. Modification of the README to include command example changes.
…z/model-zoo into xray_classifier_port
for more information, see https://pre-commit.ci
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the contribution. Would be great to add corresponding test here:
https://github.com/Project-MONAI/model-zoo/tree/dev/ci/unit_tests
large_files: | ||
- path: "models/autoencoder.pth" | ||
url: "https://drive.google.com/uc?export=download&id=1paDN1m-Q_Oy8d_BanPkRTi3RlNB_Sv_h" | ||
hash_val: "" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please also include the hash val here.
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json", | ||
"version": "1.0.0", | ||
"changelog": { | ||
"0.2": "Flipped images fixed" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should also include "1.0.0" here?
For metadata, adjusted the versions, setting current to Initial Release. I also removed nibabel from the optional package requirements as this CXR produces JPG and does not deal with nifti files.
…z/model-zoo into xray_classifier_port
@@ -0,0 +1,110 @@ | |||
{ | |||
"imports": [ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @virginiafdez , thanks for the PR. Could you also help to add a test file? The purpose is to ensure the bundle is runnable, and here is an example:
https://github.com/Project-MONAI/model-zoo/blob/dev/ci/unit_tests/test_spleen_ct_segmentation.py#L37
…erify_bundle to pass the check for model.pt, since the requirement for two models (autoencoder and diffusion_model) makes sense for them to keep their specific names. Modification of inference.json to add dummy attributes to pass the ConfigWorkflow check. Modification of large_files.yml so that models are .pt and not .pth.
…erify_bundle to pass the check for model.pt, since the requirement for two models (autoencoder and diffusion_model) makes sense for them to keep their specific names. Modification of inference.json to add dummy attributes to pass the ConfigWorkflow check. Modification of large_files.yml so that models are .pt and not .pth.
for more information, see https://pre-commit.ci
It still fails on the verify_bundle because I can only write one exception and there are two models that need to be downloaded that are not named model.pt. |
Add LICENSE file.
Modification of diffusion_model.pt name to model.pt to go through the verify_bundle Python function.
Description
As part of the porting of MONAI Generative Models, the models present in the model zoo from that repo (https://github.com/Project-MONAI/GenerativeModels/tree/main/model-zoo) need to be ported into the main model zoo from MONAI.
This is one of the models, a latent diffusion model that generates X-Rays.
Status
**Ready
Please ensure all the checkboxes:
./runtests.sh --codeformat
.CONTRIBUTING.md
).monai
,pytorch
andnumpy
are correct inmetadata.json
.eval_metrics
of the provided weights and TorchScript modules.large_file.yml
./home/your_name/
for"bundle_root"
).