Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature Request: Support for custom backend models (or documentation) #145

Open
FletcherFT opened this issue Aug 26, 2024 · 1 comment
Open

Comments

@FletcherFT
Copy link

I've been directed towards RootPainter by colleagues in marine science. I'm considering which labeling tool suite to use in my lab, and I quite like this project because of the support for folks with biology backgrounds. When comparing to other open source tools (label-studio and CVAT in particular) I'm hesitant to adopt RootPainter simply because of the fixed model architecture (U-Net) that is implemented in the backend appears to be dry.

I can't see an easy way to implement a different model supported within the torch backend. If I'm wrong about that, then I think there just needs to be better documentation on supporting other model architectures.

I understand it wouldn't be an easy thing to refactor RootPainter to be model agnostic, but I think it will future-proof RootPainter a bit if the model architecture can be easily updated to keep up with state-of-the-art.

@Abe404
Copy link
Owner

Abe404 commented Aug 27, 2024

Thanks for looking into RootPainter.

Right now the model architecture is fixed, but it should not be that hard to change it. I'm working on evaluating some more recent architectures at the moment but I don't yet have a timeline for when it will be released as part of RootPainter.

Do you have any specific new models/architectures in mind?

Eventually I'd like to support https://onnx.ai but it isn't there yet.

Kind regards,
Abraham

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants
@Abe404 @FletcherFT and others