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

Added --rmlabel and --rm-if-overlaps. Also changed the behavior of --split so that

Browse Source 
it simply partitions the data and is an invertible operation.
This commit is contained in:
Davis King 2016-09-26 20:50:54 -04:00
parent 104008b81c
commit 28a5512b64
1 changed files with 85 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
tools/imglab/src/main.cpp
View File

@ -98,7 +98,7 @@ int split_dataset (
for (unsigned long i = 0; i < data.images.size(); ++i)
{
dlib::image_dataset_metadata::image temp = data.images[i];
auto&& temp = data.images[i];
bool has_the_label = false;
// check for the label we are looking for
@ -112,24 +112,9 @@ int split_dataset (
}
if (has_the_label)
{
std::vector<dlib::image_dataset_metadata::box> boxes;
// remove other labels
for (unsigned long j = 0; j < temp.boxes.size(); ++j)
{
if (temp.boxes[j].label == label)
{
// put only the boxes with the label we want into boxes
boxes.push_back(temp.boxes[j]);
}
}
temp.boxes = boxes;
data_with.images.push_back(temp);
}
else
{
data_without.images.push_back(temp);
}
}
@ -760,9 +745,7 @@ int main(int argc, char** argv)
parser.add_option("shuffle","Randomly shuffle the order of the images listed in file <arg>.");
parser.add_option("seed", "When using --shuffle, set the random seed to the string <arg>.",1);
parser.add_option("split", "Split the contents of an XML file into two separate files. One containing the "
"images with objects labeled <arg> and another file with all the other images. Additionally, the file "
"containing the <arg> labeled objects will not contain any other labels other than <arg>. "
"That is, the images in the first file are stripped of all labels other than the <arg> labels.",1);
"images with objects labeled <arg> and another file with all the other images. ",1);
parser.add_option("add", "Add the image metadata from <arg1> into <arg2>. If any of the image "
"tags are in both files then the ones in <arg2> are deleted and replaced with the "
"image tags from <arg1>. The results are saved into merged.xml and neither <arg1> or "
@ -774,6 +757,8 @@ int main(int argc, char** argv)
parser.add_option("cluster", "Cluster all the objects in an XML file into <arg> different clusters and save "
"the results as cluster_###.xml and cluster_###.jpg files.",1);
parser.add_option("ignore", "Mark boxes labeled as <arg> as ignored. The resulting XML file is output as a separate file and the original is not modified.",1);
parser.add_option("rmlabel","Remove all boxes labeled <arg> and save the results to a new XML file.",1);
parser.add_option("rm-if-overlaps","Remove all boxes labeled <arg> if they overlap any box not labeled <arg> and save the results to a new XML file.",1);
parser.set_group_name("Cropping sub images");
parser.add_option("resample", "Crop out images that are centered on each object in the dataset. "
@ -790,7 +775,7 @@ int main(int argc, char** argv)
const char* singles[] = {"h","c","r","l","files","convert","parts","rmdiff", "rmtrunc", "rmdupes", "seed", "shuffle", "split", "add",
"flip", "rotate", "tile", "size", "cluster", "resample", "extract-chips", "min-object-size", "rmempty",
"crop-size", "cropped-object-size"};
"crop-size", "cropped-object-size", "rmlabel", "rm-if-overlaps"};
parser.check_one_time_options(singles);
const char* c_sub_ops[] = {"r", "convert"};
parser.check_sub_options("c", c_sub_ops);
@ -803,6 +788,8 @@ int main(int argc, char** argv)
parser.check_incompatible_options("c", "files");
parser.check_incompatible_options("c", "rmdiff");
parser.check_incompatible_options("c", "rmempty");
parser.check_incompatible_options("c", "rmlabel");
parser.check_incompatible_options("c", "rm-if-overlaps");
parser.check_incompatible_options("c", "rmdupes");
parser.check_incompatible_options("c", "rmtrunc");
parser.check_incompatible_options("c", "add");
@ -857,6 +844,10 @@ int main(int argc, char** argv)
parser.check_incompatible_options("rmdiff", "ignore");
parser.check_incompatible_options("rmempty", "ignore");
parser.check_incompatible_options("rmempty", "rename");
parser.check_incompatible_options("rmlabel", "ignore");
parser.check_incompatible_options("rmlabel", "rename");
parser.check_incompatible_options("rm-if-overlaps", "ignore");
parser.check_incompatible_options("rm-if-overlaps", "rename");
parser.check_incompatible_options("rmdupes", "rename");
parser.check_incompatible_options("rmdupes", "ignore");
parser.check_incompatible_options("rmtrunc", "rename");
@ -994,6 +985,80 @@ int main(int argc, char** argv)
return EXIT_SUCCESS;
}
if (parser.option("rmlabel"))
{
if (parser.number_of_arguments() != 1)
{
cerr << "The --rmlabel option requires you to give one XML file on the command line." << endl;
return EXIT_FAILURE;
}
dlib::image_dataset_metadata::dataset data;
load_image_dataset_metadata(data, parser[0]);
const auto label = parser.option("rmlabel").argument();
for (auto&& img : data.images)
{
std::vector<dlib::image_dataset_metadata::box> boxes;
for (auto&& b : img.boxes)
{
if (b.label != label)
boxes.push_back(b);
}
img.boxes = boxes;
}
save_image_dataset_metadata(data, parser[0] + ".rmlabel-"+label+".xml");
return EXIT_SUCCESS;
}
if (parser.option("rm-if-overlaps"))
{
if (parser.number_of_arguments() != 1)
{
cerr << "The --rm-if-overlaps option requires you to give one XML file on the command line." << endl;
return EXIT_FAILURE;
}
dlib::image_dataset_metadata::dataset data;
load_image_dataset_metadata(data, parser[0]);
const auto label = parser.option("rm-if-overlaps").argument();
test_box_overlap overlaps;
for (auto&& img : data.images)
{
std::vector<dlib::image_dataset_metadata::box> boxes;
for (auto&& b : img.boxes)
{
if (b.label != label)
{
boxes.push_back(b);
}
else
{
bool has_overlap = false;
for (auto&& b2 : img.boxes)
{
if (b2.label != label && overlaps(b2.rect, b.rect))
{
has_overlap = true;
break;
}
}
if (!has_overlap)
boxes.push_back(b);
}
}
img.boxes = boxes;
}
save_image_dataset_metadata(data, parser[0] + ".rm-if-overlaps-"+label+".xml");
return EXIT_SUCCESS;
}
if (parser.option("rmdupes"))
{
if (parser.number_of_arguments() != 1)