Joseph Redmon
2016-05-12 054e2b1954aafb15b0e983180dda309cfd5d831f
src/detection_layer.c
@@ -50,7 +50,7 @@
            int index = b*l.inputs;
            for (i = 0; i < locations; ++i) {
                int offset = i*l.classes;
                softmax_array(l.output + index + offset, l.classes,
                softmax_array(l.output + index + offset, l.classes, 1,
                        l.output + index + offset);
            }
            int offset = locations*l.classes;
@@ -146,7 +146,7 @@
                }
                float iou  = box_iou(out, truth);
                //printf("%d", best_index);
                //printf("%d,", best_index);
                int p_index = index + locations*l.classes + i*l.n + best_index;
                *(l.cost) -= l.noobject_scale * pow(l.output[p_index], 2);
                *(l.cost) += l.object_scale * pow(1-l.output[p_index], 2);