Separate mode for inference

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

Separate mode for inference #23

Closed YipengHu closed 4 years ago

YipengHu commented 4 years ago

Update for inference, including the fixed moving statistics in batch_norm. Also see issue raised in #22.

YipengHu commented 4 years ago

This will be handled in the new version with TF2.x