Skip to content

Commit

Permalink
Add ResNet 50
Browse files Browse the repository at this point in the history
  • Loading branch information
MarvinTeichmann committed Mar 21, 2017
1 parent 1616860 commit 65a3e12
Show file tree
Hide file tree
Showing 2 changed files with 77 additions and 2 deletions.
4 changes: 2 additions & 2 deletions hypes/KittiSeg_ResNet.json
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
},

"arch": {
"layers": 50,
"layers": 101,
"num_classes" : 2,
"image_size" : 50,
"weight": [1, 2],
Expand Down Expand Up @@ -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
}
75 changes: 75 additions & 0 deletions hypes/KittiSeg_ResNet50.json
Original file line number Diff line number Diff line change
@@ -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
}

0 comments on commit 65a3e12

Please sign in to comment.