From 65a3e1250bc304262ffa646972c0830ea5ed4006 Mon Sep 17 00:00:00 2001 From: Marvin Teichmann Date: Tue, 21 Mar 2017 09:06:09 -0400 Subject: [PATCH] Add ResNet 50 --- hypes/KittiSeg_ResNet.json | 4 +- hypes/KittiSeg_ResNet50.json | 75 ++++++++++++++++++++++++++++++++++++ 2 files changed, 77 insertions(+), 2 deletions(-) create mode 100644 hypes/KittiSeg_ResNet50.json diff --git a/hypes/KittiSeg_ResNet.json b/hypes/KittiSeg_ResNet.json index 310891a..6a8b4f0 100644 --- a/hypes/KittiSeg_ResNet.json +++ b/hypes/KittiSeg_ResNet.json @@ -19,7 +19,7 @@ }, "arch": { - "layers": 50, + "layers": 101, "num_classes" : 2, "image_size" : 50, "weight": [1, 2], @@ -70,6 +70,6 @@ "clip_norm" : 1.0, "wd": 5e-4, "load_pretrained": true, - "use_moving_average_bn": false, + "use_moving_average_bn": true, "scale_down": 0.5 } diff --git a/hypes/KittiSeg_ResNet50.json b/hypes/KittiSeg_ResNet50.json new file mode 100644 index 0000000..f7d42cd --- /dev/null +++ b/hypes/KittiSeg_ResNet50.json @@ -0,0 +1,75 @@ +{ + "model": { + "input_file": "../inputs/kitti_seg_input.py", + "architecture_file" : "../encoder/resnet.py", + "objective_file" : "../decoder/fcn.py", + "optimizer_file" : "../optimizer/generic_optimizer.py", + "evaluator_file" : "../evals/kitti_eval.py" + }, + + "path": ["../incl"], + + "data": { + "train_file" : "data_road/train3.txt", + "val_file" : "data_road/val3.txt", + "road_color" : [255,0,255], + "background_color" : [255,0,0], + "vgg_url": "https://dl.dropboxusercontent.com/u/50333326/vgg16.npy", + "kitti_url": "" + }, + + "arch": { + "layers": 50, + "num_classes" : 2, + "image_size" : 50, + "weight": [1, 2], + "num_channels" : 3, + "whitening": false + }, + + + "jitter": { + "random_resize": false, + "lower_size": 0.4, + "upper_size": 1.7, + "sig": 0.15, + "res_chance": 0.4, + "random_crop": true, + "crop_patch": false, + "patch_height": 256, + "patch_width": 256, + "max_crop": 32, + "crop_chance": 0.8, + "fix_shape": false, + "reseize_image": true, + "image_height" : 384, + "image_width" : 1248, + "augment_level": 1 + }, + + "logging": { + "display_iter": 50, + "eval_iter": 250, + "write_iter": 250, + "save_iter": 2000, + "image_iter": 20000 + }, + + "solver": { + "opt": "Adam", + "batch_size": 1, + "epsilon": 0.000000001, + "adam_eps": 0.00001, + "threads": 4, + "learning_rate": 1e-5, + "learning_rate_step": null, + "max_steps": 12000 + }, + "use_fc_wd": true, + "loss": "xentropy", + "clip_norm" : 1.0, + "wd": 5e-4, + "load_pretrained": true, + "use_moving_average_bn": true, + "scale_down": 0.5 +}