Commit 24082349 authored by Davis King's avatar Davis King

setup code to show box overlays

parent eca923f9
...@@ -252,6 +252,22 @@ on_lb_images_clicked( ...@@ -252,6 +252,22 @@ on_lb_images_clicked(
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
std::vector<dlib::image_display::overlay_rect> get_overlays (
const dlib::image_dataset_metadata::image& data
)
{
std::vector<dlib::image_display::overlay_rect> temp(data.boxes.size());
for (unsigned long i = 0; i < temp.size(); ++i)
{
temp[i].rect = data.boxes[i].rect;
temp[i].label = data.boxes[i].label;
assign_pixel(temp[i].color, rgb_pixel(255,0,0));
}
return temp;
}
// ----------------------------------------------------------------------------------------
void metadata_editor:: void metadata_editor::
load_image( load_image(
unsigned long idx unsigned long idx
...@@ -273,6 +289,7 @@ load_image( ...@@ -273,6 +289,7 @@ load_image(
} }
display.set_image(img); display.set_image(img);
display.add_overlay(get_overlays(metadata.images[idx]));
} }
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
...@@ -315,6 +332,7 @@ load_image_and_set_size( ...@@ -315,6 +332,7 @@ load_image_and_set_size(
display.set_image(img); display.set_image(img);
display.add_overlay(get_overlays(metadata.images[idx]));
} }
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment