JeVois  1.17
JeVois Smart Embedded Machine Vision Toolkit
Share this page:
PostProcessorClassify.H
Go to the documentation of this file.
1 // ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////
2 //
3 // JeVois Smart Embedded Machine Vision Toolkit - Copyright (C) 2021 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 <jevois/Types/ObjReco.H>
22 
23 namespace jevois
24 {
25  namespace dnn
26  {
27  //! Post-Processor for neural network pipeline
28  /*! This is the last step in a deep neural network processing Pipeline. \ingroup dnn */
30  public Parameter<postprocessor::classoffset, postprocessor::classes,
31  postprocessor::top, postprocessor::thresh, postprocessor::softmax,
32  postprocessor::scorescale>
33  {
34  public:
35 
36  //! Inherited constructor ok
37  using PostProcessor::PostProcessor;
38 
39  //! Destructor
40  virtual ~PostProcessorClassify();
41 
42  //! Freeze/unfreeze parameters that users should not change while running
43  void freeze(bool doit) override;
44 
45  //! Process outputs and draw/send some results
46  void process(std::vector<cv::Mat> const & outs, PreProcessor * preproc) override;
47 
48  //! Report what happened in last process() to console/output video/GUI
49  void report(jevois::StdModule * mod, jevois::RawImage * outimg = nullptr,
50  jevois::OptGUIhelper * helper = nullptr, bool overlay = true, bool idle = false) override;
51 
52  protected:
53  void onParamChange(postprocessor::classes const & param, std::string const & val) override;
54  std::map<int, std::string> itsLabels; //!< Mapping from object ID to class name
55 
56  private:
57  std::vector<jevois::ObjReco> itsObjRec;
58  };
59 
60  } // namespace dnn
61 } // namespace jevois
jevois::dnn::PostProcessorClassify::report
void report(jevois::StdModule *mod, jevois::RawImage *outimg=nullptr, jevois::OptGUIhelper *helper=nullptr, bool overlay=true, bool idle=false) override
Report what happened in last process() to console/output video/GUI.
Definition: PostProcessorClassify.C:82
jevois::RawImage
A raw image as coming from a V4L2 Camera and/or being sent out to a USB Gadget.
Definition: RawImage.H:110
jevois::dnn::PostProcessorClassify::onParamChange
void onParamChange(postprocessor::classes const &param, std::string const &val) override
Definition: PostProcessorClassify.C:37
jevois::GUIhelper
Helper class to assist modules in creating graphical and GUI elements.
Definition: GUIhelper.H:108
jevois
Definition: Concepts.dox:1
jevois::dnn::PreProcessor
Pre-Processor for neural network pipeline.
Definition: PreProcessor.H:76
jevois::dnn::PostProcessorClassify::freeze
void freeze(bool doit) override
Freeze/unfreeze parameters that users should not change while running.
Definition: PostProcessorClassify.C:31
jevois::dnn::PostProcessorClassify
Post-Processor for neural network pipeline.
Definition: PostProcessorClassify.H:29
PostProcessor.H
jevois::dnn::PostProcessorClassify::~PostProcessorClassify
virtual ~PostProcessorClassify()
Destructor.
Definition: PostProcessorClassify.C:27
jevois::dnn::PostProcessorClassify::itsLabels
std::map< int, std::string > itsLabels
Mapping from object ID to class name.
Definition: PostProcessorClassify.H:54
ObjReco.H
jevois::dnn::PostProcessor
Post-Processor for neural network pipeline.
Definition: PostProcessor.H:114
jevois::dnn::PostProcessorClassify::process
void process(std::vector< cv::Mat > const &outs, PreProcessor *preproc) override
Process outputs and draw/send some results.
Definition: PostProcessorClassify.C:51
jevois::StdModule
Base class for a module that supports standardized serial messages.
Definition: Module.H:238