cyclomon / 3dbraingen

Official Pytorch Implementation of "Generation of 3D Brain MRI Using Auto-Encoding Generative Adversarial Network" (accepted by MICCAI 2019)
MIT License
126 stars 38 forks source link

128 instead of 64 #17

Open Kunjal1999 opened 1 year ago

Kunjal1999 commented 1 year ago

Hello, I am trying to use the framework on a custom dataset. In VAEGAN_ADNI_train, I want to resize the images to 128128128 by setting sp_size=128 instead of sp_size=64. On running the code for training, for the line mean,logvar,code = E(real_images), I get the following exception:

RuntimeError                              Traceback (most recent call last)
<ipython-input-43-3a5c933a2738> in <cell line: 11>()
     19         #real_array.append(real_images)
     20         z_rand = Variable(torch.randn((_batch_size, latent_dim)),requires_grad=False).cuda()
---> 21         mean,logvar,code = E(real_images)
     22         x_rec = G(code)
     23         x_rand = G(z_rand)

5 frames
/usr/local/lib/python3.10/dist-packages/torch/nn/modules/linear.py in forward(self, input)
    112 
    113     def forward(self, input: Tensor) -> Tensor:
--> 114         return F.linear(input, self.weight, self.bias)
    115 
    116     def extra_repr(self) -> str:

RuntimeError: mat1 and mat2 shapes cannot be multiplied (4x262144 and 32768x2048)

What other changes do I need to make to make the framework accept images of various sizes? (64,128,142,256, etc.)

cyclomon commented 1 year ago

Hi, since we did not implemented the model on different resolution,

the model should be modified to get the higher resolution images.

I apologize for inconvenience.

Thanks

Kunjal1999 commented 1 year ago

Could you kindly provide more specific details regarding the modifications needed to get higher resolution images?