File size: 5,412 Bytes
9375c9a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 |
// Copyright (C) 2011 Davis E. King ([email protected])
// License: Boost Software License See LICENSE.txt for the full license.
#ifndef DLIB_IMAGE_DAtASET_METADATA_Hh_
#define DLIB_IMAGE_DAtASET_METADATA_Hh_
#include <string>
#include <vector>
#include "../geometry.h"
// ----------------------------------------------------------------------------------------
namespace dlib
{
namespace image_dataset_metadata
{
// ------------------------------------------------------------------------------------
enum gender_t
{
UNKNOWN,
MALE,
FEMALE
};
// ------------------------------------------------------------------------------------
struct box
{
/*!
WHAT THIS OBJECT REPRESENTS
This object represents an annotated rectangular area of an image.
It is typically used to mark the location of an object such as a
person, car, etc.
The main variable of interest is rect. It gives the location of
the box. All the other variables are optional.
!*/
box(
) :
difficult(false),
truncated(false),
occluded(false),
ignore(false),
pose(0),
detection_score(0),
angle(0),
gender(UNKNOWN),
age(0)
{}
box (
const rectangle& rect_
) :
rect(rect_),
difficult(false),
truncated(false),
occluded(false),
ignore(false),
pose(0),
detection_score(0),
angle(0),
gender(UNKNOWN),
age(0)
{}
rectangle rect;
std::map<std::string,point> parts;
// optional fields
std::string label;
bool difficult;
bool truncated;
bool occluded;
bool ignore;
double pose;
double detection_score;
// The angle of the object in radians. Positive values indicate that the
// object at the center of the box is rotated clockwise by angle radians. A
// value of 0 would indicate that the object is in its "standard" upright pose.
// Therefore, to make the object appear upright we would have to rotate the
// image counter-clockwise by angle radians.
double angle;
gender_t gender;
double age;
bool has_label() const { return label.size() != 0; }
/*!
ensures
- returns true if label metadata is present and false otherwise.
!*/
};
// ------------------------------------------------------------------------------------
struct image
{
/*!
WHAT THIS OBJECT REPRESENTS
This object represents an annotated image.
!*/
image() {}
image(const std::string& f) : filename(f) {}
std::string filename;
std::vector<box> boxes;
};
// ------------------------------------------------------------------------------------
struct dataset
{
/*!
WHAT THIS OBJECT REPRESENTS
This object represents a labeled set of images. In particular, it
contains the filename for each image as well as annotated boxes.
!*/
std::vector<image> images;
std::string comment;
std::string name;
};
// ------------------------------------------------------------------------------------
void save_image_dataset_metadata (
const dataset& meta,
const std::string& filename
);
/*!
ensures
- Writes the contents of the meta object to a file with the given
filename. The file will be in an XML format.
throws
- dlib::error
This exception is thrown if there is an error which prevents
this function from succeeding.
!*/
// ------------------------------------------------------------------------------------
void load_image_dataset_metadata (
dataset& meta,
const std::string& filename
);
/*!
ensures
- Attempts to interpret filename as a file containing XML formatted data
as produced by the save_image_dataset_metadata() function. Then
meta is loaded with the contents of the file.
throws
- dlib::error
This exception is thrown if there is an error which prevents
this function from succeeding.
!*/
// ------------------------------------------------------------------------------------
}
}
// ----------------------------------------------------------------------------------------
#ifdef NO_MAKEFILE
#include "image_dataset_metadata.cpp"
#endif
#endif // DLIB_IMAGE_DAtASET_METADATA_Hh_
|