HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Few-Shot Object Detection and Viewpoint Estimation for Objects in the Wild

Yang Xiao; Vincent Lepetit; Renaud Marlet

Few-Shot Object Detection and Viewpoint Estimation for Objects in the Wild

Abstract

Detecting objects and estimating their viewpoints in images are key tasks of 3D scene understanding. Recent approaches have achieved excellent results on very large benchmarks for object detection and viewpoint estimation. However, performances are still lagging behind for novel object categories with few samples. In this paper, we tackle the problems of few-shot object detection and few-shot viewpoint estimation. We demonstrate on both tasks the benefits of guiding the network prediction with class-representative features extracted from data in different modalities: image patches for object detection, and aligned 3D models for viewpoint estimation. Despite its simplicity, our method outperforms state-of-the-art methods by a large margin on a range of datasets, including PASCAL and COCO for few-shot object detection, and Pascal3D+ and ObjectNet3D for few-shot viewpoint estimation. Furthermore, when the 3D model is not available, we introduce a simple category-agnostic viewpoint estimation method by exploiting geometrical similarities and consistent pose labelling across different classes. While it moderately reduces performance, this approach still obtains better results than previous methods in this setting. Last, for the first time, we tackle the combination of both few-shot tasks, on three challenging benchmarks for viewpoint estimation in the wild, ObjectNet3D, Pascal3D+ and Pix3D, showing very promising results.

Code Repositories

YoungXIAO13/FewShotDetection
pytorch
Mentioned in GitHub
YoungXIAO13/PoseContrast
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
few-shot-object-detection-on-ms-coco-10-shotFSDetView
AP: 12.5
few-shot-object-detection-on-ms-coco-30-shotFsDetView
AP: 14.7

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Few-Shot Object Detection and Viewpoint Estimation for Objects in the Wild | Papers | HyperAI