weecology / weecology/DeepForest

Polygon model support

Open
#758 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
774
Forks
265
PR merge metrics
No merged PRs in 30d

Description

As part of milestone 2.0, we want to support polygon prediction and modeling. Here is a roadmap

I expect that the user will not have to declare polygons anywhere, when they choose mask-rcnn in the config, or when we read_file we will know that they have polygon data. We may need to add a class attribute to deepforest.main that keeps track of the current geometry.

Optional ideas

  • Compare output of mask RCNN with segment anything based on the DeepForest box model. see #460

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reviewing the models/ directory, the dataset implementation in deepforest/dataset.py, and the MillionTrees TreePolygons.py example to understand the proposed mask output. Then inspect preprocess.split_raster and the visualization utilities for polygon compatibility. Done would include mask-rcnn support, a polygon dataset class, compatible utilities, and an example notebook using polygon data.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
computer-vision, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.