Skip to content

Code for our Reproducibility Challenge Paper "On end-to-end 6DoF object pose estimation and robustness to object scale"

Notifications You must be signed in to change notification settings

tzole1155/EndToEndObjectPose

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

EndToEndObjectPose

| OpenReview | This repository contrains our code for replicating the following papers:

  • End-to-End Learnable Geometric Vision by Backpropagating PnP Optimization paper
  • HigherHRNet: Scale-Aware Representation Learning for Bottom-Up Human Pose Estimation paper

OnEndToEnd method overview
Overview of the approach

Setup

Our implementation is based on moai. Please follow the intructions for installing the package.

Data

The used dataset for running the experiments is UAVA. Please follow the instructions for downloading the data.

Configuration files

We provide the configuration files for training the models reported in the paper. All the configuration files for conducting the reported experiments are split in unique folders containing all the necessary parts. More specifically each folder contains:

  • coordinate_regression: Which is the main configuration file consisting of the definition of the different components.
  • data: configuration file for defining the data loaders.
  • model: configuration file for defining the model and differenet components.
  • losses: configuration file for defining the losses.
  • metrics: configuration file for defining the evaluation metrics.
  • options: configuration file to define the visualization functions, logging parameters, etc.

Training

To start a training, run

moai train --coordinate_regression.yaml <main config file> --config-dir . <path_to_configuration_files>
root=<data_path>
obj_file=<drone_obj_file>
metadata=<metadata_path>

About

Code for our Reproducibility Challenge Paper "On end-to-end 6DoF object pose estimation and robustness to object scale"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published