← all repositories

ucbdrive/few-shot-object-detection

Few-shot object detection framework implementing the TFA two-stage fine-tuning approach from ICML 2020.

1.2k stars Python Computer VisionML Frameworks
few-shot-object-detection
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

FsDet provides implementations for few-shot object detection benchmarks on PASCAL VOC, COCO, and LVIS datasets. The repository includes a two-stage fine-tuning approach where an object detector is first trained on base classes with abundant data, then only the last layers are fine-tuned on a small balanced dataset of novel classes. It offers benchmark results, pre-trained models, and modularized code for adding custom datasets and models.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.