代码地址: https://github.com/makefile/darknet
编译要求: VS2013 update5 及其之后的版本(低版本对C++标准支持较差)
配置opencv来显示图片结果,如果不配置OpenCV,则支持的图片类型较少,结果将直接保存到文件.
下载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 *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版移植源码在 这里 .
Darknet的OpenMP加速在多个for循环中,代码在这,在windows上运行时有时出现访问冲突.按照(这里的代码)修改即可,使用一个for循环多次调用原各个gemm函数,在for前边加#pragma omp parallel for语句.
可能出现名字冲突等,需要改一些变量的名字.
将框架的初始化和模型的加载,网络预测等操作封装到了一个类中,使用__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.