Image prediction

[Li-En-Good]

Hi,

Thanks for sharing this amazing work!
I'm wondering if this can be modified to perform image prediction from a slice of image to a stack (predict signal intensity)?
If it is possible, could you share how you would do it?

[Omloujain]

@Li-En-Good Please have you tried this code on google colab? it requires GPU? I need response please