facebookresearch / facebookresearch/detectron2

[Feature Request] Dataset cache for COCO annotations that don't fit in memory

Open
#4,184 0 comments 2 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
34.7k
Forks
7.9k
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature
I'd like to request adding [D2Go's cache feature](https://github.com/facebookresearch/d2go/blob/87374efb134e539090e0b5c476809dc35bf6aedb/d2go/data/config.py#L39) to the main Detecton2 repo.

## Motivation & Examples

This feature would make it easier to train on very large datasets.

Currently it's extremely cumbersome to train and perform coco evaluation on a custom format other than coco (for example a TorchData iterable), though coco format annotations may not fit into memory for very large datasets.

Contributor guide

Open the contributing guide

Research direction

Start by reading the linked D2Go cache implementation in d2go/data/config.py, then trace Detectron2's dataset loading and COCO evaluation paths for where annotations are held in memory. Done means large COCO-annotation datasets can be cached and used for training and evaluation without requiring all annotations to fit in memory.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
computer-vision, data
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.