Skip to content

Commit

Permalink
Merge pull request #3 from torrvision/master
Browse files Browse the repository at this point in the history
Update TVG_CRFRNN_traintest_example.prototxt
  • Loading branch information
bittnt committed Apr 12, 2016
2 parents e53097b + 82fb2e9 commit 650e988
Showing 1 changed file with 5 additions and 2 deletions.
Expand Up @@ -124,7 +124,8 @@ layers { bottom: 'fc7' top: 'fc7' name: 'drop7' type: DROPOUT
layers { name: 'score-fr' type: CONVOLUTION bottom: 'fc7' top: 'score'
blobs_lr: 1 blobs_lr: 2 weight_decay: 1 weight_decay: 0
convolution_param { engine: CAFFE num_output: 21 kernel_size: 1 } }


#warning: if you change the num_output in deconvolution, please make sure you initialize the caffemodel weights for this properly!
layers { type: DECONVOLUTION name: 'score2' bottom: 'score' top: 'score2'
blobs_lr: 1 blobs_lr: 2 weight_decay: 1 weight_decay: 0
convolution_param { kernel_size: 4 stride: 2 num_output: 21 } }
Expand All @@ -140,6 +141,7 @@ layers { type: ELTWISE name: 'fuse' bottom: 'score2' bottom: 'score-pool4c'
top: 'score-fused'
eltwise_param { operation: SUM } }

#warning: if you change the num_output in deconvolution, please make sure you initialize the caffemodel weights for this properly!
layers { type: DECONVOLUTION name: 'score4' bottom: 'score-fused'
top: 'score4'
blobs_lr: 1 weight_decay: 1
Expand All @@ -155,7 +157,8 @@ layers { type: CROP name: 'crop' bottom: 'score-pool3' bottom: 'score4'
layers { type: ELTWISE name: 'fuse' bottom: 'score4' bottom: 'score-pool3c'
top: 'score-final'
eltwise_param { operation: SUM } }


#warning: if you change the num_output in deconvolution, please make sure you initialize the caffemodel weights for this properly!
layers { type: DECONVOLUTION name: 'upsample'
bottom: 'score-final' top: 'bigscore'
blobs_lr: 0
Expand Down

0 comments on commit 650e988

Please sign in to comment.