Fixed GPU compile flag

This commit is contained in:
Matt Hill
2015-03-12 22:58:02 -04:00
parent 27a2ae7145
commit 7f693373d5
4 changed files with 8 additions and 8 deletions

View File

@@ -31,10 +31,6 @@
#define DEFAULT_CONFIG_FILE "/etc/openalpr/openalpr.conf"
#endif
#ifndef COMPILE_GPU
#define COMPILE_GPU 0
#endif
#define ENV_VARIABLE_CONFIG_FILE "OPENALPR_CONFIG_FILE"
#endif // OPENALPR_CONSTANTS_H

View File

@@ -17,14 +17,14 @@
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "detectorcuda.h"
#if COMPILE_GPU
using namespace cv;
using namespace std;
#ifdef COMPILE_GPU
namespace alpr
{

View File

@@ -20,7 +20,6 @@
#ifndef OPENALPR_DETECTORCUDA_H
#define OPENALPR_DETECTORCUDA_H
#if COMPILE_GPU
#include <stdio.h>
#include <iostream>
@@ -33,6 +32,10 @@
#include "opencv2/gpu/gpu.hpp"
#include "detector.h"
#include "detectorcpu.h"
#ifdef COMPILE_GPU
namespace alpr
{

View File

@@ -11,12 +11,13 @@ namespace alpr
}
else if (config->gpu_mode == 1)
{
#if COMPILE_GPU
#ifdef COMPILE_GPU
return new DetectorCUDA(config);
#else
std::cerr << "Error: GPU detector requested, but GPU extensions are not compiled. " <<
"Add COMPILE_GPU=1 to the compiler definitions to enable GPU compilation." <<
std::endl;
return new DetectorCPU(config);
#endif
}
}