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 age and gender fields to image metadata.

Browse Source
This commit is contained in:
Davis King 2017-03-21 21:05:21 -04:00
parent 2dfbf778e6
commit 0a8d94add4
2 changed files with 36 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
dlib/data_io/image_dataset_metadata.cpp
View File

@ -80,6 +80,12 @@ namespace dlib
fout << " ignore='" << b.ignore << "'";
if (b.angle != 0)
fout << " angle='" << b.angle << "'";
if (b.age != 0)
fout << " age='" << b.age << "'";
if (b.gender == FEMALE)
fout << " gender='female'";
else if (b.gender == MALE)
fout << " gender='male'";
if (b.pose != 0)
fout << " pose='" << b.pose << "'";
if (b.detection_score != 0)
@ -196,6 +202,18 @@ namespace dlib
if (atts.is_in_list("occluded")) temp_box.occluded = sa = atts["occluded"];
if (atts.is_in_list("ignore")) temp_box.ignore = sa = atts["ignore"];
if (atts.is_in_list("angle")) temp_box.angle = sa = atts["angle"];
if (atts.is_in_list("age")) temp_box.age = sa = atts["age"];
if (atts.is_in_list("gender"))
{
if (atts["gender"] == "male")
temp_box.gender = MALE;
else if (atts["gender"] == "female")
temp_box.gender = FEMALE;
else if (atts["gender"] == "unknown")
temp_box.gender = UNKNOWN;
else
throw dlib::error("Invalid gender string in box attribute.");
}
if (atts.is_in_list("pose")) temp_box.pose = sa = atts["pose"];
if (atts.is_in_list("detection_score")) temp_box.detection_score = sa = atts["detection_score"];

20
dlib/data_io/image_dataset_metadata.h
View File

@ -14,6 +14,15 @@ namespace dlib
namespace image_dataset_metadata
{
// ------------------------------------------------------------------------------------
enum gender_t
{
UNKNOWN,
MALE,
FEMALE
};
// ------------------------------------------------------------------------------------
struct box
@ -36,7 +45,9 @@ namespace dlib
ignore(false),
pose(0),
detection_score(0),
angle(0)
angle(0),
gender(UNKNOWN),
age(0)
{}
box (
@ -49,7 +60,9 @@ namespace dlib
ignore(false),
pose(0),
detection_score(0),
angle(0)
angle(0),
gender(UNKNOWN),
age(0)
{}
rectangle rect;
@ -72,6 +85,9 @@ namespace dlib
// image counter-clockwise by angle radians.
double angle;
gender_t gender;
double age;
bool has_label() const { return label.size() != 0; }
/*!
ensures