Generative Model with Coordinate Metric Learning for Object Recognition Based on 3D Models

TL;DR

This paper introduces a generative model with metric learning that leverages synthetic images from 3D models to improve object recognition, reducing data collection efforts and enhancing recognition accuracy across varied conditions.

Contribution

The paper proposes a novel generative model combining Bayesian rendering and metric learning with a coordinate training strategy for effective synthetic-to-real object recognition.

Findings

Achieved over 50% accuracy on ShapeNet database.

Reduced data collection workload using synthetic images.

Enhanced recognition robustness across different poses and lighting.

Abstract

Given large amount of real photos for training, Convolutional neural network shows excellent performance on object recognition tasks. However, the process of collecting data is so tedious and the background are also limited which makes it hard to establish a perfect database. In this paper, our generative model trained with synthetic images rendered from 3D models reduces the workload of data collection and limitation of conditions. Our structure is composed of two sub-networks: semantic foreground object reconstruction network based on Bayesian inference and classification network based on multi-triplet cost function for avoiding over-fitting problem on monotone surface and fully utilizing pose information by establishing sphere-like distribution of descriptors in each category which is helpful for recognition on regular photos according to poses, lighting condition, background and category information of rendered images. Firstly, our conjugate structure called generative model with metric learning utilizing additional foreground object channels generated from Bayesian rendering as the joint of two sub-networks. Multi-triplet cost function based on poses for object recognition are used for metric learning which makes it possible training a category classifier purely based on synthetic data. Secondly, we design a coordinate training strategy with the help of adaptive noises acting as corruption on input images to help both sub-networks benefit from each other and avoid inharmonious parameter tuning due to different convergence speed of two sub-networks. Our structure achieves the state of the art accuracy of over 50\% on ShapeNet database with data migration obstacle from synthetic images to real photos. This pipeline makes it applicable to do recognition on real images only based on 3D models.