Newer
Older
SC-SfMLearner_for_NLab / prediction / pred_pose.py
@planck planck on 6 Dec 2020 680 bytes readmeの更新
import sys, os

sys.path.append(os.path.abspath(".."))

import torch
import cv2
import numpy as np
import pandas as pd
import argparse
import models

parser = argparse.ArgumentParser(description='Structure from Motion Learner',
                                 formatter_class=argparse.ArgumentDefaultsHelpFormatter)

parser.add_argument('--weight_pth', required=True, help="path to disp model's pth file")
parser.add_argument('--path_to_yaml', default="../datasets/data_for_SC_SfMLearner/environment.yaml", help="path to environent")
parser.add_argument('--use_camera', type=int, default=-9999, help="カメラを使うならポート番号を指定")
args = parser.parse_args()