JeVoisBase  1.3
JeVois Smart Embedded Machine Vision Toolkit Base Modules
Share this page:
ObjectRecognitionMNIST.H
Go to the documentation of this file.
1 // ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
2 //
3 // JeVois Smart Embedded Machine Vision Toolkit - Copyright (C) 2016 by Laurent Itti, the University of Southern
4 // California (USC), and iLab at USC. See http://iLab.usc.edu and http://jevois.org for information about this project.
5 //
6 // This file is part of the JeVois Smart Embedded Machine Vision Toolkit. This program is free software; you can
7 // redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software
8 // Foundation, version 2. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
9 // without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
10 // License for more details. You should have received a copy of the GNU General Public License along with this program;
11 // if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
12 //
13 // Contact information: Laurent Itti - 3641 Watt Way, HNB-07A - Los Angeles, CA 90089-2520 - USA.
14 // Tel: +1 213 740 3527 - itti@pollux.usc.edu - http://iLab.usc.edu - http://jevois.org
15 // ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
16 /*! \file */
17 
18 #pragma once
19 
21 #include "tiny-dnn/tiny_dnn/nodes.h"
22 
23 //! Object recognition CNN for MNIST handwritten digits
24 /*! \ingroup components */
25 class ObjectRecognitionMNIST : public ObjectRecognition<tiny_dnn::sequential>
26 {
27  public:
28  //! Constructor, loads the given CNN, its sizes must match our (fixed) internal network structure
29  /*! All network data is assumed to be in the module's path plus "tiny-dnn/<instance>". In there, we will look for
30  weights.tnn, and if not found, we will train the network using data in that path and then save weights.tnn. */
31  ObjectRecognitionMNIST(std::string const & instance);
32 
33  //! Destructor
34  virtual ~ObjectRecognitionMNIST();
35 
36  //! Define the network structure
37  virtual void define() override;
38 
39  //! Train the network
40  virtual void train(std::string const & path) override;
41 
42  //! Return the name of a given category (0-based index in the vector of results)
43  virtual std::string const & category(size_t idx) const override;
44 };
virtual void define() override
Define the network structure.
virtual ~ObjectRecognitionMNIST()
Destructor.
ObjectRecognitionMNIST(std::string const &instance)
Constructor, loads the given CNN, its sizes must match our (fixed) internal network structure...
virtual void train(std::string const &path) override
Train the network.
Wrapper around a neural network implemented by with the tiny-dnn framework by Taiga Nomi...
virtual std::string const & category(size_t idx) const override
Return the name of a given category (0-based index in the vector of results)
Object recognition CNN for MNIST handwritten digits.