From f7781be0fd23230c53a9e03add25a20855fbb15b Mon Sep 17 00:00:00 2001 From: Manu Mathew Date: Tue, 28 Jan 2020 18:12:43 +0530 Subject: [PATCH] release commit --- run_depth.sh | 4 ++-- scripts/train_depth_main.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/run_depth.sh b/run_depth.sh index f96d1af..861b294 100755 --- a/run_depth.sh +++ b/run_depth.sh @@ -5,8 +5,8 @@ ## Training ## ===================================================================================== #### KITTI Depth (Manual Download) - Training with MobileNetV2+DeeplabV3Lite -#python ./scripts/train_depth_main.py --dataset_name kitti_depth --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \ -#--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth +python ./scripts/train_depth_main.py --dataset_name kitti_depth --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \ +--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth #### KITTI Depth (Manual Download) - Training with ResNet50+FPN #python ./scripts/train_depth_main.py --dataset_name kitti_depth --model_name fpn_pixel2pixel_aspp_resnet50 --data_path ./data/datasets/kitti/kitti_depth/data --img_resize 384 768 --output_size 374 1242 \ diff --git a/scripts/train_depth_main.py b/scripts/train_depth_main.py index f0f7e78..e5db906 100755 --- a/scripts/train_depth_main.py +++ b/scripts/train_depth_main.py @@ -83,8 +83,8 @@ args.pretrained = './data/modelzoo/semantic_segmentation/cityscapes/deeplabv3lit args.model_config.input_channels = (3,) # [3,3] args.model_config.output_type = ['depth'] args.model_config.output_channels = [1] -args.model_config.output_range = [(0,64)] # important note: set this output_range parameter in the inference script as well - # this is an important difference from the semantic segmentation script. +args.model_config.output_range = [(0,128)] # important note: set this output_range parameter in the inference script as well + # this is an important difference from the semantic segmentation script. args.losses = [['supervised_loss', 'scale_loss', 'supervised_error_var']] #[['supervised_loss', 'scale_loss']] args.loss_mult_factors = [[0.125, 0.125, 4.0]] -- 2.39.2