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

Create a main.deepforest.set_label function #935

Open
bw4sz opened this issue Feb 25, 2025 · 0 comments
Open

Create a main.deepforest.set_label function #935

bw4sz opened this issue Feb 25, 2025 · 0 comments
Labels
good first issue Good for newcomers

Comments

@bw4sz
Copy link
Collaborator

bw4sz commented Feb 25, 2025

Most users won't know that you need to reset both the id -> class and class -> id dictionary.

m = main.deepforest()
m.load_model("weecology/deepforest-bird")
m.label_dict = {"Object":0}
m.numeric_to_label_dict = {0:"Object"}

we should have

m.set_labels({"Object":0})

that should reset both. If not m.evaluate will say the label is called "bird" from the loaded model weights.

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

No branches or pull requests

1 participant