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

Point detection workflow #809

Open
7 tasks
bw4sz opened this issue Oct 14, 2024 · 0 comments
Open
7 tasks

Point detection workflow #809

bw4sz opened this issue Oct 14, 2024 · 0 comments

Comments

@bw4sz
Copy link
Collaborator

bw4sz commented Oct 14, 2024

As part of DeepForest 2.0 we need a model for point prediction.

A couple examples we can work from

https://github.com/xingyizhou/CenterNet.
https://github.com/FateScript/CenterNet-better
https://arxiv.org/abs/1904.07850

We have the data from

https://doi.org/10.1016/j.jag.2024.103848

Roadmap

  • Add small point dataset example to get_data. We should use the existing box dataset (OSBS_029) and just get the centroids
  • Add a point dataloader to datasets.py similar to
    class BoundingBoxDataset(Dataset):
  • Add a point model to models.
  • Example training script in examples/
  • Confirm visualization
  • Confirm can be extended to multi-class

@Mu-Magdy might be interested, since its isolated from the code base, doesn't have a time frame and fits objectives.

Bonus

  • Train a tree point detector on the MillionTrees point datasets and/or the center points of the boxes/polygons datasets.
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

1 participant