search

BrainLesion / AURORA

Segmentation models for cancer metastasis in brain MR
https://www.thegreenjournal.com/article/S0167-8140(23)89795-X/fulltext
Apache License 2.0
5 stars 1 forks source link

fixes a bug that occurs when trying to open the downloaded model weights #50

Closed LennartPhil closed 6 months ago

LennartPhil commented 6 months ago

Fixes the bug that occurs when trying to open the downloaded model weights.

Considering that the model weights are named like t1-o_best.tar, the _load_model function should look for the value of the interference_mode and of the model_selection

MarcelRosier commented 6 months ago

without .value it worked just fine for me (internally python will call the enums str method which should result in the same thing?). So I'm curious: What Python version are you using to run into an issue? However, being explicit is probably a good thing here - so thanks for contributing!

LennartPhil commented 6 months ago

I'm currently on Python 3.11, glad I could help!