Darknet windows移植(YOLO v2)

Darknet windows移植

代码地址: https://github.com/makefile/darknet

编译要求: VS2013 update5 及其之后的版本(低版本对C++标准支持较差)

配置opencv来显示图片结果,如果不配置OpenCV,则支持的图片类型较少,结果将直接保存到文件.

pthread库

下载windows版pthread库,将头文件与编译好的lib与dll文件挑出来供Darknet使用.在VS配置中添加pthreadVC2.lib.

时间函数

linux下的时间函数库#include <sys/time.h>需要被替换.可以用clock()来替换what_time_is_it_now(),或者如下代码来定义what_time_is_it_now:

#include "gettimeofday.h"
#define CLOCK_REALTIME 0
#pragma warning(disable: 4996) //disable deprecated warning as error
#define BILLION                             (1E9)

static BOOL g_first_time = 1;
static LARGE_INTEGER g_counts_per_sec;

int clock_gettime(int dummy, struct timespec *ct)
{
	LARGE_INTEGER count;

	if (g_first_time)
	{
		g_first_time = 0;

		if (0 == QueryPerformanceFrequency(&g_counts_per_sec))
		{
			g_counts_per_sec.QuadPart = 0;
		}
	}

	if ((NULL == ct) || (g_counts_per_sec.QuadPart <= 0) ||
		(0 == QueryPerformanceCounter(&count)))
	{
		return -1;
	}

	ct->tv_sec = count.QuadPart / g_counts_per_sec.QuadPart;
	ct->tv_nsec = ((count.QuadPart % g_counts_per_sec.QuadPart) * BILLION) / g_counts_per_sec.QuadPart;

	return 0;
}

double what_time_is_it_now()
{
    struct timespec now;
    clock_gettime(CLOCK_REALTIME, &now);
    return now.tv_sec + now.tv_nsec*1e-9;
}

gettimeofday.h与gettimeofday.c实现在 这里 .

void *指针加法

访问地址时对void *arr进行加法:arr+偏移数字时VS编译不通过,强制类型转换为char*可通过且没有错误.

动态数组

static const int nind = 2;之后创建了大小了nind 的数组,VS的低版本可能不支持arr[变量]这样的数组定义.

替换成宏定义define nind 2,也可简单将数组定义为arr[2].

其它函数定义,修饰符,头文件

#if defined(_MSC_VER) && _MSC_VER < 1900
	#define inline __inline
	#define snprintf(buf,len, format,...) _snprintf_s(buf, len,len, format, __VA_ARGS__)
#endif

#ifdef _WIN32
	#define popen	_popen
	#define pclose	_pclose
	//#include <windows.h>
	#define sleep(secs)	Sleep(secs * 1000) //windows.h millsecs
#endif

getopt.h与unistd.h均为非标准C的linux API,windows版移植源码在 这里 .

OpenMP

Darknet的OpenMP加速在多个for循环中,代码在这,在windows上运行时有时出现访问冲突.按照(这里的代码)修改即可,使用一个for循环多次调用原各个gemm函数,在for前边加#pragma omp parallel for语句.

VS的其它小毛病

可能出现名字冲突等,需要改一些变量的名字.

导出动态链接库

将框架的初始化和模型的加载,网络预测等操作封装到了一个类中,使用__declspec(dllexport)修饰符来将函数导出到dll中.头文件与lib,dll文件可供其它Windows项目使用.

代码在这里:yolo_v2_class.hpp,yolo_v2_class.cpp,yolo_console_dll.cpp.

使用测试

运行命令:darknet.exe detector test data/coco.data yolo.cfg yolo.weights -i 0 -thresh 0.2

大小1176x510的图片,测试耗时4min,开启OpenMP加速后60s (i7,8线程).

其它加速方法:使用其它的库,如MKL等.参考https://groups.google.com/forum/#!topic/darknet/AZ-jp45bDpI.