3DRDN-CycleGAN icon indicating copy to clipboard operation
3DRDN-CycleGAN copied to clipboard

Inference

Open dengyijia214 opened this issue 1 year ago • 3 comments

Could you pleaes help me with the inference problem, I have no idea of how to inference the model which I have trained well. I could not find the ckpt.meta file.

Thank you !

dengyijia214 avatar Apr 26 '23 06:04 dengyijia214

Hi @dengyijia214

In any GAN model the inference is done using the generator after the training is complete. You can use the this function to automatically load the latest generator saved during training.

And then you can easily do inference with it like in this line of code. Make sure the input has 4 dimensions (number_of_images, x_size, y_szie, z_size).

Let me know if any part needs more clarification.

omagdy avatar Apr 26 '23 08:04 omagdy

Thanks for your reply, I got sick these days so I did not reply you on time. I have not describe my question clearly, I want to know if I want to get the whole predicted image, for example the whole 512512300 size CT, is there anything I could do?

dengyijia214 avatar May 10 '23 02:05 dengyijia214

Hi @dengyijia214

You would have to split the whole image into smaller blocks using a package like this one and then group them back together.

omagdy avatar May 23 '23 18:05 omagdy