Add option to specify object classes list file
[tidl/tidl-api.git] / examples / common / object_classes.h
1 /******************************************************************************
2  * Copyright (c) 2018, Texas Instruments Incorporated - http://www.ti.com/
3  *   All rights reserved.
4  *
5  *   Redistribution and use in source and binary forms, with or without
6  *   modification, are permitted provided that the following conditions are met:
7  *       * Redistributions of source code must retain the above copyright
8  *         notice, this list of conditions and the following disclaimer.
9  *       * Redistributions in binary form must reproduce the above copyright
10  *         notice, this list of conditions and the following disclaimer in the
11  *         documentation and/or other materials provided with the distribution.
12  *       * Neither the name of Texas Instruments Incorporated nor the
13  *         names of its contributors may be used to endorse or promote products
14  *         derived from this software without specific prior written permission.
15  *
16  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17  *   AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  *   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  *   ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
20  *   LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21  *   CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22  *   SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23  *   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24  *   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25  *   ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
26  *   THE POSSIBILITY OF SUCH DAMAGE.
27  *****************************************************************************/
29 #ifndef _OBJECT_CLASSES_H_
30 #define _OBJECT_CLASSES_H_
32 #include <string>
33 #include <vector>
35 struct ObjectClass {
36     std::string       label;
37     struct {
38         unsigned char blue;
39         unsigned char green;
40         unsigned char red;
41     } color;
43     ObjectClass(const std::string& l, unsigned char b, unsigned char g,
44                 unsigned char r) : label { l }, color { b, g, r} {}
45 };
47 class ObjectClasses {
48   public:
49     ObjectClasses(const std::string& json_file);
50     const ObjectClass& At(unsigned int index);
52     unsigned int       GetNumClasses()  { return num_classes_m; }
53     const std::string& GetNetworkName() { return network_name_m; }
55   private:
56     unsigned int              num_classes_m;
57     std::string               network_name_m;
58     std::vector<ObjectClass>  classes_m;
59 };
61 #endif