×

darknetlib Darknet的C API库

消耗积分:0 | 格式:zip | 大小:1.08 MB | 2022-06-22

Petc

分享资料个

授权协议 MIT
开发语言 C/C++
操作系统 Windows
软件类型 开源软件

软件简介

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();
}

运行结果:

poYBAGKn6EmAU-o7AABfaCv2mS8900.jpg

 

声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉

评论(0)
发评论

下载排行榜

全部0条评论

快来发表一下你的评论吧 !