mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Changed small_object

Browse Source
This commit is contained in:
AlexeyAB
2018-03-30 14:46:51 +03:00
parent 9fbd1936eb
commit 23cb35e6c8
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/data.c
View File

@ -292,7 +292,8 @@ void fill_truth_region(char *path, float *truth, int classes, int num_boxes, int
free(boxes); 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, int small_object) 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, int net_w, int net_h)
{ {
char labelpath[4096]; char labelpath[4096];
find_replace(path, "images", "labels", labelpath); find_replace(path, "images", "labels", labelpath);
@ -307,9 +308,12 @@ void fill_truth_detection(char *path, int num_boxes, float *truth, int classes,
int i; int i;
box_label *boxes = read_boxes(labelpath, &count); box_label *boxes = read_boxes(labelpath, &count);
if (small_object == 1) { if (small_object == 1) {
float lowest_w = 1.F / net_w;
float lowest_h = 1.F / net_h;
printf(" lowest_w = %f, lowest_h = %f \n", lowest_w, lowest_h);
for (i = 0; i < count; ++i) { for (i = 0; i < count; ++i) {
if (boxes[i].w < 0.001F) boxes[i].w = 0.001F; if (boxes[i].w < lowest_w) boxes[i].w = lowest_w;
if (boxes[i].h < 0.001F) boxes[i].h = 0.001F; if (boxes[i].h < lowest_h) boxes[i].h = lowest_h;
} }
} }
randomize_boxes(boxes, count); randomize_boxes(boxes, count);
@ -738,7 +742,7 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int boxes, in
//show_image(ai, "aug"); //show_image(ai, "aug");
//cvWaitKey(0); //cvWaitKey(0);
fill_truth_detection(filename, boxes, d.y.vals[i], classes, flip, dx, dy, 1./sx, 1./sy, small_object); fill_truth_detection(filename, boxes, d.y.vals[i], classes, flip, dx, dy, 1./sx, 1./sy, small_object, w, h);
cvReleaseImage(&src); cvReleaseImage(&src);
} }
@ -789,7 +793,7 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int boxes, in
random_distort_image(sized, hue, saturation, exposure); random_distort_image(sized, hue, saturation, exposure);
d.X.vals[i] = sized.data; 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, small_object); fill_truth_detection(random_paths[i], boxes, d.y.vals[i], classes, flip, dx, dy, 1. / sx, 1. / sy, small_object, w, h);
free_image(orig); free_image(orig);
free_image(cropped); free_image(cropped);