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

Now the tool can modify the ground truth boxes.

Browse Source
This commit is contained in:
Davis King 2011-06-20 20:25:06 -04:00
parent 0a1069d0a7
commit 92ac70dc6d
2 changed files with 37 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tools/imglab/src/metadata_editor.cpp
View File

@ -49,6 +49,9 @@ metadata_editor(
overlay_label_name.set_text("Next Label: ");
overlay_label.set_width(200);
display.set_overlay_rects_changed_handler(*this, &metadata_editor::on_overlay_rects_changed);
overlay_label.set_text_modified_handler(*this, &metadata_editor::on_overlay_label_changed);
mbar.set_number_of_menus(1);
mbar.set_menu_name(0,"File",'F');
@ -350,4 +353,36 @@ load_image_and_set_size(
// ----------------------------------------------------------------------------------------
void metadata_editor::
on_overlay_rects_changed(
)
{
using namespace dlib::image_dataset_metadata;
if (image_pos < metadata.images.size())
{
const std::vector<image_display::overlay_rect>& rects = display.get_overlay_rects();
std::vector<box>& boxes = metadata.images[image_pos].boxes;
boxes.clear();
for (unsigned long i = 0; i < rects.size(); ++i)
{
box temp;
temp.label = rects[i].label;
temp.rect = rects[i].rect;
boxes.push_back(temp);
}
}
}
// ----------------------------------------------------------------------------------------
void metadata_editor::
on_overlay_label_changed(
)
{
display.set_default_overlay_rect_label(trim(overlay_label.text()));
}
// ----------------------------------------------------------------------------------------

2
tools/imglab/src/metadata_editor.h
View File

@ -35,6 +35,8 @@ private:
void save_metadata_to_file (const std::string& file);
void load_image(unsigned long idx);
void load_image_and_set_size(unsigned long idx);
void on_overlay_rects_changed();
void on_overlay_label_changed();
std::string filename;
dlib::image_dataset_metadata::dataset metadata;