juglab / n2v

This is the implementation of Noise2Void training.
Other
406 stars 110 forks source link

Fix prediction axes #38

Closed tibuch closed 5 years ago

tibuch commented 5 years ago

Move channel axis to last position in model.predict(), because the normalization expects channel to be last. Resolves issue #37.