NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ignore even small instances

Browse Source
This commit is contained in:
Juha Reunanen 2019-10-26 20:55:11 +03:00
parent 5fd90407a9
commit 075d24afdd
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/dnn_instance_segmentation_train_ex.cpp
View File

@ -470,9 +470,17 @@ int ignore_overlapped_boxes(
std::vector<mmod_rect> load_mmod_rects(const image_info& image_info) std::vector<mmod_rect> load_mmod_rects(const image_info& image_info)
{ {
matrix<rgb_pixel> rgb_label_image; matrix<rgb_pixel> rgb_label_image;
load_image(rgb_label_image, image_info.label_filename); load_image(rgb_label_image, image_info.label_filename);
auto mmod_rects = rgb_label_image_to_mmod_rects(rgb_label_image); auto mmod_rects = rgb_label_image_to_mmod_rects(rgb_label_image);
ignore_overlapped_boxes(mmod_rects, test_box_overlap(0.50, 0.95)); ignore_overlapped_boxes(mmod_rects, test_box_overlap(0.50, 0.95));
for (auto& rect : mmod_rects)
if (rect.rect.width() < 35 && rect.rect.height() < 35)
rect.ignore = true;
return mmod_rects; return mmod_rects;
}; };