From a1af57d8d60b50e8188f36b7f74752c8cc124177 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 15 Feb 2018 12:43:25 +0000 Subject: [PATCH] Added C implementation of calculation mAP (mean average precision) using Darknet --- src/cuda.h | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/cuda.h b/src/cuda.h index 31f9092..31577ff 100644 --- a/src/cuda.h +++ b/src/cuda.h @@ -26,6 +26,7 @@ void cuda_push_array(float *x_gpu, float *x, size_t n); void cuda_pull_array(float *x_gpu, float *x, size_t n); void cuda_set_device(int n); +int cuda_get_device(); void cuda_free(float *x_gpu); void cuda_random(float *x_gpu, size_t n); float cuda_compare(float *x_gpu, float *x, size_t n, char *s); -- Gitblit v1.10.0