YipengHu / label-reg

(This repo is no longer up-to-date. Any updates will be at https://github.com/DeepRegNet/DeepReg/) A demo of the re-factored label-driven registration code, based on "Weakly-supervised convolutional neural networks for multimodal image registration"
Apache License 2.0
117 stars 32 forks source link

The detail Running Environment #21

Closed NanYoMy closed 5 years ago

NanYoMy commented 5 years ago

It's a really great job! I am trying to run my own labeled 3D-image data on the Nvidia 1080GPU.
And I found the 3D-CNN is really time and memory consuming. could you please tell me the size of 3D-image that you had fed into the neural network. Thx

YipengHu commented 5 years ago

Yep it can be of a problem. See this: https://github.com/YipengHu/label-reg/wiki/Memory-Constraint