LOST  by valeoai

Unsupervised object discovery and detection framework

Created 4 years ago
255 stars

Top 98.8% on SourcePulse

GitHubView on GitHub
Project Summary

LOST (Localizing Objects with Self-Supervised Transformers and no Labels) provides a PyTorch implementation for unsupervised object discovery. It targets researchers and engineers seeking to identify and localize objects without labeled data, serving as a crucial step towards fully unsupervised object detection by leveraging novel self-supervised transformer approaches.

How It Works

The core approach utilizes self-supervised transformers, building upon the DINO framework. LOST learns object localization by analyzing visual features derived from self-supervision, bypassing the need for manual annotations. This enables the generation of object maps and bounding boxes, which can then train downstream object detectors without supervision.

Quick Start & Requirements

  • Installation: Clone repo, pip install -r requirements.txt. Requires separate DINO framework installation (commit ba9edd1 recommended). Detectron2 (v0.5) needed for object detection extensions.
  • Prerequisites: Python 3.7, PyTorch 1.7.1, CUDA 10.2.
  • Datasets: PASCAL VOC07/12 and COCO datasets required in datasets folder.
  • Links: Paper: arXiv, DINO: GitHub.

Highlighted Details

  • Achieves competitive corloc results: VOC07 (61.9), VOC12 (64.0), COCO20k (50.7) with ViT-S/16 and DINO pre-training.
  • Extends to unsupervised object detection (Class-Agnostic/Aware) using LOST predictions as pseudo-GT with detectron2.
  • Performance varies across ViT/ResNet architectures, DINO/ImageNet pre-training, and patch sizes, detailed in provided tables.

Maintenance & Community

No specific community channels (Discord, Slack) or ongoing maintenance details are provided in the README.

Licensing & Compatibility

Apache 2.0 license. Permissive for commercial use and integration into closed-source projects.

Limitations & Caveats

Strict version dependencies exist for Python (3.7), PyTorch (1.7.1), CUDA (10.2), DINO (specific commit), and detectron2 (v0.5), posing potential compatibility challenges. Setup requires downloading large datasets and complex environment configuration.

Health Check
Last Commit

2 years ago

Responsiveness

Inactive

Pull Requests (30d)
0
Issues (30d)
0
Star History
5 stars in the last 30 days

Explore Similar Projects

Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems") and Luis Capelo Luis Capelo(Cofounder of Lightning AI).

GroundingDINO by IDEA-Research

0.4%
9k
Object detection via grounded pre-training research paper
Created 2 years ago
Updated 1 year ago
Starred by Andrej Karpathy Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), Deshraj Yadav Deshraj Yadav(Cofounder of Mem0), and
7 more.

rcnn by rbgirshick

0.1%
2k
Object detection system using CNNs and region proposals
Created 11 years ago
Updated 8 years ago
Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems") and Kevin Hou Kevin Hou(Head of Product Engineering at Windsurf).

ImageAI by OlafenwaMoses

0.0%
9k
Python library for computer vision tasks
Created 7 years ago
Updated 1 year ago
Feedback? Help us improve.