/PoseNet_TUM

PoseNet for Camera Relocalization

Primary LanguageJupyter Notebook

UMich EECS 545 Winter 2020 Project: Camera Relocalization using PoseNet-50 on TUM Dataset

Exploring localization task on TMU dataset

NOTE: This repository is a modified from Apolloscape dataset for localization task.

Read original blog post about PoseNet implementation details https://capsulesbot.com/blog/2018/08/24/apolloscape-posenet-pytorch.html

Download the dataset here : https://vision.in.tum.de/data/datasets/mono-dataset
Use TUM_PoseNet_Train.ipynb to train your model.
You can download pre-train model under checkpoint folder.
After downloading the model and dataset, TUM_PoseNet_Eval.ipynb can be executed.