HealthML / active-segmentation

ActiveSegmentation: A Simulation Framework for Benchmarking Active Learning Strategies for 3D Medical Image Segmentation
GNU Affero General Public License v3.0
16 stars 1 forks source link

Fix loss weighting #141

Closed josafatburmeister closed 2 years ago

josafatburmeister commented 2 years ago

Closes #137

Fixes setting of loss weights in Pytorch U-net model. Everything else needed for #137 is already implemented.