| | |
|---|
| | | box_label *boxes = read_boxes(labelpath, &count); |
|---|
| | | if (small_object == 1) { |
|---|
| | | for (i = 0; i < count; ++i) { |
|---|
| | | if (boxes[i].w < 0.01) boxes[i].w = 0.01; |
|---|
| | | if (boxes[i].h < 0.01) boxes[i].h = 0.01; |
|---|
| | | if (boxes[i].w < 0.001F) boxes[i].w = 0.001F; |
|---|
| | | if (boxes[i].h < 0.001F) boxes[i].h = 0.001F; |
|---|
| | | } |
|---|
| | | } |
|---|
| | | randomize_boxes(boxes, count); |
|---|
| | |
|---|
| | | id = boxes[i].id; |
|---|
| | | |
|---|
| | | // not detect small objects |
|---|
| | | if ((w < 0.001 || h < 0.001)) continue; |
|---|
| | | if ((w < 0.001F || h < 0.001F)) continue; |
|---|
| | | |
|---|
| | | truth[i*5+0] = x; |
|---|
| | | truth[i*5+1] = y; |
|---|
| | |
|---|
| | | } else if (a.type == IMAGE_DATA){ |
|---|
| | | *(a.im) = load_image_color(a.path, 0, 0); |
|---|
| | | *(a.resized) = resize_image(*(a.im), a.w, a.h); |
|---|
| | | }else if (a.type == LETTERBOX_DATA) { |
|---|
| | | *(a.im) = load_image_color(a.path, 0, 0); |
|---|
| | | *(a.resized) = letterbox_image(*(a.im), a.w, a.h); |
|---|
| | | } else if (a.type == TAG_DATA){ |
|---|
| | | *a.d = load_data_tag(a.paths, a.n, a.m, a.classes, a.min, a.max, a.size, a.angle, a.aspect, a.hue, a.saturation, a.exposure); |
|---|
| | | } |
|---|
