代码学习 (1) :Unsupervised Monocular Depth Estimation with Left-Right Consistency

代码学习笔记

Unsupervised Monocular Depth Estimation with Left-Right Consistency :monodepth.main
源码:monodepth

"""

  代码学习 注释专用
  song
  stay hungry stay foolish

"""


from __future__ import absolute_import, division, print_function  # 完成 python2 和 python3 之间的转换,放在文件首部
"""
分别引入的是 python3 中的 : absolute_import:绝对引入;division:整除;print_function: 输出
"""

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'  # only keep warnings and errors
"""
import os  
os.environ["TF_CPP_MIN_LOG_LEVEL"]='1' # 这是默认的显示等级,显示所有信息  
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只显示 warning 和 Error   
os.environ["TF_CPP_MIN_LOG_LEVEL"]='3' # 只显示 Error
"""

import numpy as np  # 处理和存储大型矩阵
import argparse  # 参数设置
import re  # 正则表达式
import time  # 提供时间功能
import tensorflow as tf
import tensorflow.contrib.slim as slim  # tensorflow的辅助工具,用来简化代码

from monodepth_model import *
from monodepth_dataloader import *
from average_gradients import *
"""
    从文件中引入所有的 函数 和 类
"""

parser = argparse.ArgumentParser(description='Monodepth TensorFlow implementation.')  # 参数的定义

parser.add_argument('--mode',                      type=str,   help='train or test', default='train')
parser.add_argument('--model_name',                type=str,   help='model name', default='monodepth')
parser.add_argument('--encoder',                   type=str,   help='type of encoder, vgg or resnet50', default='vgg')
parser.add_argument('--dataset',                   type=str,   help='dataset to train on, kitti, or cityscapes', default='kitti')
parser.add_argument('--data_path',                 type=str,   help='path to the data', required=True)
parser.add_argument('--filenames_file',            type=str,   help='path to the filenames text file', required=True)
parser.add_argument('--input_height',              type=int,   help='input height', default=256)
parser.add_argument('--input_width',               type=int,   help='input width', default=512)
parser.add_argument('--batch_size',                type=int,   help='batch size', default=8)
parser.add_argument('--num_epochs',                type=int,   help='number of epochs', default=50)
parser.add_argument('--learning_rate',             type=float, help='initial learning rate', default=1e-4)
parser.add_argument('--lr_loss_weight',            type=float, help='left-right consistency weight', default=1.0)
parser.add_argument('--alpha_image_loss',          type=float, help='weight between SSIM and L1 in the image loss', default=0.85)
parser.add_argument('--disp_gradient_loss_weight', type=float, help='disparity smoothness weigth', default
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值