YOLO2源码分析——读取训练图片[by zhangzexuan]
回去补了一些知识,加上自己又懒,所以空了一段时间,才又回来看看YOLO的代码……
废话不多说,继续看代码。
这次分析下YOLO是如何读取训练图片的,同时来看看YOLO在读取到训练图片后都对它们做了些什么。
训练入口以及图片路径的读取我就不做记录了,直接看通过路径读取图片的部分吧。
//data.c[944-957]
data load_data_detection(int n, char **paths, int m, int w, int h, int boxes, int classes, float jitter, float hue, float saturation, float exposure)
{
char **random_paths = get_random_paths(paths, n, m);
int i;
data d = {
0};
d.shallow = 0;
d.X.rows = n;
d.X.vals = calloc(d.X.rows, sizeof(float*));
d.X.cols = h*w*3;
d.y = make_matrix(n, 5*boxes);
for(i = 0; i < n; ++i){