It takes into account the Difficult for calculating mAP for PascalVOC

This commit is contained in:
AlexeyAB
2018-02-16 23:55:37 +03:00
parent a1af57d8d6
commit 65bff2683b
9 changed files with 179 additions and 20 deletions

View File

@ -35,7 +35,7 @@ void smooth_l1_cpu(int n, float *pred, float *truth, float *delta, float *error)
void l2_cpu(int n, float *pred, float *truth, float *delta, float *error);
void weighted_sum_cpu(float *a, float *b, float *s, int num, float *c);
void softmax(float *input, int n, float temp, float *output);
void softmax(float *input, int n, float temp, float *output, int stride);
#ifdef GPU
#include "cuda.h"