Allow napari GUI to accept custom weights - Githubissues

brainglobe / cellfinder

Automated 3D cell detection in very large images

https://brainglobe.info/documentation/cellfinder/index.html

BSD 3-Clause "New" or "Revised" License

181 stars 39 forks source link

Allow napari GUI to accept custom weights #460

Closed IgorTatarnikov closed 1 month ago

IgorTatarnikov commented 1 month ago

Before submitting a pull request (PR), please read the contributing guide.

Please fill out as much of this template as you can, but if you have any problems or questions, just leave a comment and we will help out :)

Description

What is this PR

[ ] Bug fix
[x] Addition of a new feature
[ ] Other

Why is this PR needed? See #459

What does this PR do? Check if the trained_model ends in .h5. If so, warn the user, and load the weights into a default model.

References

closes #459

How has this PR been tested?

Tested locally with the model weights from the paper.

Is this a breaking change?

No.

Does this PR require an update to the documentation?

Yes, will come in a subsequent PR.

Checklist:

[x] The code has been tested locally
[ ] Tests have been added to cover all new functionality (unit & integration)
[ ] The documentation has been updated to reflect any changes
[x] The code has been formatted with pre-commit