Fixed im2col mistake >< face#palm

This commit is contained in:
Joseph Redmon
2015-03-26 19:13:59 -07:00
parent e92f7d301c
commit d7d7da2653
7 changed files with 34 additions and 13 deletions

View File

@ -93,6 +93,19 @@ void forward_detection_layer(const detection_layer layer, network_state state)
}
}
/*
int count = 0;
for(i = 0; i < layer.batch*locations; ++i){
for(j = 0; j < layer.classes+layer.background; ++j){
printf("%f, ", layer.output[count++]);
}
printf("\n");
for(j = 0; j < layer.coords; ++j){
printf("%f, ", layer.output[count++]);
}
printf("\n");
}
*/
/*
if(layer.background || 1){
for(i = 0; i < layer.batch*locations; ++i){
int index = i*(layer.classes+layer.coords+layer.background);
@ -123,8 +136,9 @@ void backward_detection_layer(const detection_layer layer, network_state state)
state.delta[in_i++] = scale*layer.delta[out_i++];
}
if (layer.nuisance) ;
else if (layer.background) gradient_array(layer.output + out_i, layer.coords, LOGISTIC, layer.delta + out_i);
if (layer.nuisance) {
}else if (layer.background) gradient_array(layer.output + out_i, layer.coords, LOGISTIC, layer.delta + out_i);
for(j = 0; j < layer.coords; ++j){
state.delta[in_i++] = layer.delta[out_i++];
}