darknetlib 是 Darknet 的 C API 库,简单地封装了 Darknet 的一些接口,主要用于为 LC-Finder 项目提供图像识别功能。
特性
- 支持简单的异常处理
- 重新定义了接口,接口命名风格参考自 LeveDB/c.h
- 适合作为动态库来使用
示例代码
#include
#include
#include
#include "../include/darknet.h"
void print_detections(const darknet_detections_t *dets)
{
size_t i, j;
darknet_detection_t *det;
printf("detections:\n");
for (i = 0; i < dets->length; ++i) {
det = &dets->list[i];
printf("[%zu] best name: %s, box: (%g, %g, %g, %g)\n", i,
det->best_name, det->box.x, det->box.y, det->box.w,
det->box.h);
printf("names: \n");
for (j = 0; j < det->names_count; ++j) {
printf("%s, %g%%\n", det->names[j],
det->prob[j] * 100.0f);
}
}
}
int detect(void)
{
clock_t c;
int code = 0;
// initialize variables so that darknet can check if they need to be
// destroyed after catching exception
darknet_config_t *cfg = NULL;
darknet_dataconfig_t *datacfg = NULL;
darknet_detections_t dets = { 0 };
darknet_detector_t *d = NULL;
darknet_network_t *net = NULL;
c = clock();
darknet_try
{
cfg = darknet_config_load("cfg/yolov3.cfg");
datacfg = darknet_dataconfig_load("cfg/coco.data");
net = darknet_network_create(cfg);
darknet_network_load_weights(net, "yolov3.weights");
d = darknet_detector_create(net, datacfg);
printf("\ntime: %.2fs\n\n",
(clock() - c) * 1.0f / CLOCKS_PER_SEC);
c = clock();
darknet_detector_test(d, "img/dog.jpg", &dets);
printf("\ntime: %.2fs\n\n",
(clock() - c) * 1.0f / CLOCKS_PER_SEC);
}
darknet_catch(err)
{
printf("error: %s\n", darknet_get_last_error_string());
code = -1;
}
darknet_etry;
print_detections(&dets);
darknet_detections_destroy(&dets);
darknet_config_destroy(cfg);
darknet_dataconfig_destroy(datacfg);
darknet_detector_destroy(d);
darknet_network_destroy(net);
return code;
}
int main(int argc, char *argv[])
{
return detect();
}
运行结果: