【论文】Unsupervised Monocular Depth Estimation with Left-Right Consistency

论文地址
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

猜你喜欢

转载自blog.csdn.net/yaoyao_chen/article/details/130414643