誰でも(SC)SfMLearnerを使えるようにした奴

@planck planck authored on 7 Dec 2020
datasets 重みファイルから深度,ポーズ推定を行う機能の追加 5 years ago
datasets_module バグの改善,prediction機能の追加 5 years ago
models train機能の実装 5 years ago
prediction predictionの更新 5 years ago
.gitignore バグの改善,prediction機能の追加 5 years ago
README.md readmeの更新 5 years ago
custom_transforms.py train機能の実装 5 years ago
inverse_warp.py train機能の実装 5 years ago
loss_functions.py train機能の実装 5 years ago
requirements.txt 重みファイルから深度,ポーズ推定を行う機能の追加 5 years ago
train.py 重みファイルから深度,ポーズ推定を行う機能の追加 5 years ago
utils.py train機能の実装 5 years ago
README.md

SC-SfMLearner_for_NLab

Unsupervised depth and pose estimationの手法のうちの一つであるSC-SfMLearner(実際にはEndo-SfMLeaner)のpytorch実装です.

使用者のオリジナルデータセットでもトレーニングや推定をコマンドライン上の操作だけですべて済むように調整しました.
参考にした論文は

Unsupervised Scale-consistent Depth and Ego-motion Learning from Monocular Video
Bian, Jia-Wang and Li, Zhichao and Wang, Naiyan and Zhan, Huangying and Shen, Chunhua and Cheng, Ming-Ming and Reid, Ian
SC-SfMLearnerの公式GitHub

EndoSLAM Dataset and An Unsupervised Monocular Visual Odometry and Depth Estimation Approach for Endoscopic Videos: Endo-SfMLearner Kutsev Bengisu Ozyoruk and Guliz Irem Gokceler and Gulfize Coskun and Kagan Incetan and Yasin Almalioglu and Faisal Mahmood and Eva Curto and Luis Perdigoto and Marina Oliveira and Hasan Sahin and Helder Araujo and Henrique Alexandrino and Nicholas J. Durr and Hunter B. Gilbert and Mehmet Turan
EndoSLAMとEndo-SfMLearner公式GitHub

demo

このコードは色々なコードの複合で成り立っているので非商用利用の用途でしか