Fixes for small objects. And fixes of densenet201_yolo.cfg, resnet50_yolo.cfg, resnet152_yolo.cfg

This commit is contained in:
AlexeyAB
2018-03-09 01:42:58 +03:00
parent 100d6f7801
commit a6c51e3b75
7 changed files with 33 additions and 52 deletions

View File

@ -918,7 +918,7 @@ void calc_anchors(char *datacfg, int num_of_clusters, int final_width, int final
float box_h = points->data.fl[i * 2 + 1];
//int cluster_idx = labels->data.i[i];
int cluster_idx = 0;
float min_dist = 1000000;
float min_dist = FLT_MAX;
for (j = 0; j < num_of_clusters; ++j) {
float anchor_w = centers->data.fl[j * 2];
float anchor_h = centers->data.fl[j * 2 + 1];