computational-cell-analytics / micro-sam

Segment Anything for Microscopy
https://computational-cell-analytics.github.io/micro-sam/
MIT License
312 stars 36 forks source link

Api cleanup #648

Closed constantinpape closed 1 week ago

constantinpape commented 1 week ago

I went ahead and cleaned up the function signatures related to both LoRA and 3D models. This will need some changes in your code to account for this (and I will wait with merging so that you can check the changes, I pinged you in some of the most important places in the code). But this now has a more unified interface, especially for the 3d models, which use the same function signatures as the original SAM model, making compatibility of the training code easier.

cc @anwai98 @lufre1 @caroteu

constantinpape commented 1 week ago

Also, I created a new submodule models and moved everything related to custom models there, + created some unit tests for them.

constantinpape commented 1 week ago

Thanks for checking everyone, I will go ahead and merge!