search

NoaCahan / X-ray2CTPA

X-ray2CTPA: Generating 3D CTPA scans from 2D X-ray conditioning
MIT License
13 stars 0 forks source link

About ddpm--------------No module named 'ddpm.discriminator'、generative et.al. #2

Open zfw-cv opened 3 weeks ago

zfw-cv commented 3 weeks ago

image Dear author, when I run the code according to your prompts, I encounter the following problem: Traceback (most recent call last): File "train_ddpm.py", line 2, in from ddpm.diffusion import Unet3D,GaussianDiffusion, Trainer File "/home//Desktop/projects/3d_medical/X-ray2CTPA-main/ddpm/init.py", line 1, in from ddpm.diffusion import Unet3D, GaussianDiffusion, Trainer File "/home//Desktop/projects/3d_medical/X-ray2CTPA-main/ddpm/diffusion.py", line 38, in from ddpm.discriminator import Discriminator, ones_target, zeros_target ModuleNotFoundError: No module named 'ddpm.discriminator' Then locate the place in the picture. However, I found that ddpm does not seem to have a discriminator and generative. I would like to ask how to solve this problem? Thank you

zfw-cv commented 3 weeks ago

I have solved it. The generative could be found in : https://github.com/Project-MONAI/GenerativeModels/tree/main And you should pip install monai package.

NoaCahan commented 3 weeks ago

Exactly. Thanks I'll update the requirements.txt file