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.

Sign up for GitHub

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

Jump to bottom

Add class weights parameter to classification/detection trainers #1287

Closed
adamjstewart opened this issue Apr 25, 2023 · 0 comments · Fixed by #1592
Closed

Add class weights parameter to classification/detection trainers #1287

adamjstewart opened this issue Apr 25, 2023 · 0 comments · Fixed by #1592
Milestone
0.5.0

Comments

@adamjstewart
Copy link
Collaborator

Summary

#1221 adds a class_weights parameter to the SemanticSegmentation trainer. We should do the same for other class-based trainers.

Rationale

Class imbalance is common in geospatial datasets.

Implementation

Copy implementation from #1221

Alternatives

No response

Additional information

No response
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
0.5.0
Development

Successfully merging a pull request may close this issue.

ClassificationTask: add class_weights parameter adamjstewart/torchgeo
1 participant
@adamjstewart