论文地址
github项目地址-Tensorflow
github项目地址-Pytorch
基于Pytorch的项目有几个bug需要调整:
#main_monodepth_pytorch.py:
def main(args): -> def main()
def return-arguments(): #当中的 data_dir, val_data_dir等前面需要加--
def train(self): # best_loss is never used.
def train(self): # 最后一行
self.save(self.args.model_path)#这个path不正确,介于模型已经保存在_last.pth 以及 _cpt.pth中,这行可以被注释
class Model:
...
else:
self.model.load_state_dict(torch.load(args.model_path)) ->
# 需要改成实际存储的地址
self.model.load_state_dict(torch.load(args.model_path[:-4] + '_last.pth'))
这篇文章一共combine了3个loss function,具体可见loss.py
中的image loss
,L-R consistency
,Disparities smoothness