azavea / azavea/raster-vision

Add ability to use different Objectdetection models than FasterRCNN

Open
#2,147 1 comment 0 reactions 0 assignees View on GitHub
feature-request
Dominant language
Python
Stars
2.2k
Forks
397
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature
It would be good to be able to use more up to date models than FasterRCNN in raster-vision.
Torchvision offers some possible dropin replacements / alternatives like RetinaNet or SSD based detectors.

## Motivation
More up to date models promise better performance at lower GPU needs.

## Pitch

The object_detection_learner in raster-vision should be able to accept a configuration parameter which allows a different model to be used, as written above.

## Alternatives

The user could do this manually in the code, but this is not very scaleable and unclear how many manual changes are needed.

## Additional context

-

Contributor guide

Open the contributing guide

Research direction

Start at the object_detection_learner entry point and review how its model is currently selected. Compare the requested RetinaNet or SSD alternatives and define a configuration path that lets users choose them; done means the learner accepts that choice and supports the selected detector.

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
Mostly clear
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.