mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Changes to make routing work better

Browse Source
This commit is contained in:
Joseph Redmon
2015-07-21 16:09:33 -07:00
parent e56d1eff13
commit d00f0a1ccd
18 changed files with 29 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/detection_layer.c
View File

@ -141,20 +141,20 @@ void backward_detection_layer(const detection_layer l, network_state state)
float scale = 1;
float latent_delta = 0;
if(l.joint) scale = state.input[in_i++];
else if (l.objectness) state.delta[in_i++] = -l.delta[out_i++];
else if (l.background) state.delta[in_i++] = scale*l.delta[out_i++];
else if (l.objectness) state.delta[in_i++] += -l.delta[out_i++];
else if (l.background) state.delta[in_i++] += scale*l.delta[out_i++];
for(j = 0; j < l.classes; ++j){
latent_delta += state.input[in_i]*l.delta[out_i];
state.delta[in_i++] = scale*l.delta[out_i++];
state.delta[in_i++] += scale*l.delta[out_i++];
}
if (l.objectness) {
}else if (l.background) gradient_array(l.output + out_i, l.coords, LOGISTIC, l.delta + out_i);
for(j = 0; j < l.coords; ++j){
state.delta[in_i++] = l.delta[out_i++];
state.delta[in_i++] += l.delta[out_i++];
}
if(l.joint) state.delta[in_i-l.coords-l.classes-l.joint] = latent_delta;
if(l.joint) state.delta[in_i-l.coords-l.classes-l.joint] += latent_delta;
}
}
@ -198,7 +198,8 @@ void backward_detection_layer_gpu(detection_layer l, network_state state)
cpu_state.truth = truth_cpu;
cpu_state.delta = delta_cpu;
cuda_pull_array(state.input, in_cpu, l.batch*l.inputs);
cuda_pull_array(state.input, in_cpu, l.batch*l.inputs);
cuda_pull_array(state.delta, delta_cpu, l.batch*l.inputs);
cuda_pull_array(l.delta_gpu, l.delta, l.batch*outputs);
backward_detection_layer(l, cpu_state);
cuda_push_array(state.delta, delta_cpu, l.batch*l.inputs);