cnn::RegCNNNumuHandler Class Reference

Wrapper for caffe::Net which handles construction and prediction. More...

#include <RegCNNNumuHandler.h>

Public Member Functions
	RegCNNNumuHandler (const fhicl::ParameterSet &pset)
	Constructor which takes a pset with DeployProto and ModelFile fields. More...
std::vector< float >	Predict (const RegPixelMap &pm, bool fLongestTrackContained)
	Return prediction arrays for RegPixelMap. More...

Private Attributes
cnn::TFRegNetHandler	fTFHandlerContained
cnn::TFRegNetHandler	fTFHandlerExiting

Detailed Description

Wrapper for caffe::Net which handles construction and prediction.

Definition at line 21 of file RegCNNNumuHandler.h.

Constructor & Destructor Documentation

cnn::RegCNNNumuHandler::RegCNNNumuHandler

(

const fhicl::ParameterSet &

pset

)

Constructor which takes a pset with DeployProto and ModelFile fields.

Definition at line 21 of file RegCNNNumuHandler.cxx.

                                                                   :
  fTFHandlerContained (pset.get<fhicl::ParameterSet> ("TFNetHandlerContained")),
  fTFHandlerExiting   (pset.get<fhicl::ParameterSet> ("TFNetHandlerExiting"))
  {
  }

Member Function Documentation

std::vector< float > cnn::RegCNNNumuHandler::Predict	(	const RegPixelMap &	pm,
		bool	fLongestTrackContained
	)

Return prediction arrays for RegPixelMap.

Definition at line 27 of file RegCNNNumuHandler.cxx.

  {
    std::vector<float> cnnResults;
    if (fLongestTrackContained) {
      cnnResults = fTFHandlerContained.Predict(pm);
    } else {
      cnnResults = fTFHandlerExiting.Predict(pm);
    }

    //std::cout << "Number of CNN result vectors " << cnnResults.size() << " with " << cnnResults[0].size() << " categories" << std::endl;

    //std::cout << "summary: ";
    //for(auto const v : cnnResults[0]){
    //  std::cout << v << ", ";
    //}
    //std::cout << std::endl;

    return cnnResults;
  }

Member Data Documentation

cnn::TFRegNetHandler cnn::RegCNNNumuHandler::fTFHandlerContained

private

Definition at line 33 of file RegCNNNumuHandler.h.

cnn::TFRegNetHandler cnn::RegCNNNumuHandler::fTFHandlerExiting

private

Definition at line 34 of file RegCNNNumuHandler.h.

---

The documentation for this class was generated from the following files:

• dunereco/dunereco/RegCNN/art/RegCNNNumuHandler.h
• dunereco/dunereco/RegCNN/art/RegCNNNumuHandler.cxx