Advertisement

darknet源码理解 (2) 图像读取

阅读量:

darknet源码解析(二)---图像的加载过程

概述:

本文主要讲解在darknet框架中,图像数据是如何被加载和处理的。

上一篇内容介绍了darknet的整体架构。

正文:

首先,在darknet.h文件中,作者定义了一个名为image的结构体,用于存储图像数据及其相关属性。该结构体包含以下几个成员变量:

int w, int h, int c, float* data

从变量名可以推测,它们分别对应图像的宽度、高度、通道数以及像素数据。

回顾前一篇文章可知,在detector.c文件中的test_detector()函数中,包含了一个完整的端到端流程。具体包括:

模型初始化、加载模型权重、读取图像、对图像进行尺寸调整、将图像输入模型进行前向计算、根据模型输出获取目标的位置和类别信息,并将这些信息标注在原始图像上。

其中,load_image_color(input, 0, 0)函数根据输入路径input读取图像数据。该函数定义在src/image.c文件中,并通过调用load_image()函数(该函数同样位于image.c文件中)来实现具体的图像加载操作。

接下来将展示load_image()函数的具体实现代码

复制代码
 image

全部评论 (0)

还没有任何评论哟~