| | |
|---|
| | | free(boxes); |
|---|
| | | } |
|---|
| | | |
|---|
| | | void fill_truth_detection(char *path, int num_boxes, float *truth, int classes, int flip, float dx, float dy, float sx, float sy) |
|---|
| | | void fill_truth_detection(char *path, int num_boxes, float *truth, int classes, int flip, float dx, float dy, float sx, float sy, int small_object) |
|---|
| | | { |
|---|
| | | char labelpath[4096]; |
|---|
| | | find_replace(path, "images", "labels", labelpath); |
|---|
| | |
|---|
| | | find_replace(labelpath, ".JPEG", ".txt", labelpath); |
|---|
| | | int count = 0; |
|---|
| | | box_label *boxes = read_boxes(labelpath, &count); |
|---|
| | | if (small_object == 1) { |
|---|
| | | for (int 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; |
|---|
| | | } |
|---|
| | | } |
|---|
| | | randomize_boxes(boxes, count); |
|---|
| | | correct_boxes(boxes, count, dx, dy, sx, sy, flip); |
|---|
| | | if(count > num_boxes) count = num_boxes; |
|---|
| | |
|---|
| | | h = boxes[i].h; |
|---|
| | | id = boxes[i].id; |
|---|
| | | |
|---|
| | | if ((w < .01 || h < .01)) continue; |
|---|
| | | // not detect small objects |
|---|
| | | if ((w < 0.01 || h < 0.01)) continue; |
|---|
| | | |
|---|
| | | truth[i*5+0] = x; |
|---|
| | | truth[i*5+1] = y; |
|---|
| | |
|---|
| | | return d; |
|---|
| | | } |
|---|
| | | |
|---|
| | | 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) |
|---|
| | | 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, int small_object) |
|---|
| | | { |
|---|
| | | char **random_paths = get_random_paths(paths, n, m); |
|---|
| | | int i; |
|---|
| | |
|---|
| | | random_distort_image(sized, hue, saturation, exposure); |
|---|
| | | d.X.vals[i] = sized.data; |
|---|
| | | |
|---|
| | | fill_truth_detection(random_paths[i], boxes, d.y.vals[i], classes, flip, dx, dy, 1./sx, 1./sy); |
|---|
| | | fill_truth_detection(random_paths[i], boxes, d.y.vals[i], classes, flip, dx, dy, 1./sx, 1./sy, small_object); |
|---|
| | | |
|---|
| | | free_image(orig); |
|---|
| | | free_image(cropped); |
|---|
| | |
|---|
| | | } else if (a.type == REGION_DATA){ |
|---|
| | | *a.d = load_data_region(a.n, a.paths, a.m, a.w, a.h, a.num_boxes, a.classes, a.jitter, a.hue, a.saturation, a.exposure); |
|---|
| | | } else if (a.type == DETECTION_DATA){ |
|---|
| | | *a.d = load_data_detection(a.n, a.paths, a.m, a.w, a.h, a.num_boxes, a.classes, a.jitter, a.hue, a.saturation, a.exposure); |
|---|
| | | *a.d = load_data_detection(a.n, a.paths, a.m, a.w, a.h, a.num_boxes, a.classes, a.jitter, a.hue, a.saturation, a.exposure, a.small_object); |
|---|
| | | } else if (a.type == SWAG_DATA){ |
|---|
| | | *a.d = load_data_swag(a.paths, a.n, a.classes, a.jitter); |
|---|
| | | } else if (a.type == COMPARE_DATA){ |
|---|
