cvn::TFNetHandler Class Reference

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

#include <TFNetHandler.h>

Public Member Functions
	TFNetHandler (const fhicl::ParameterSet &pset)
	Constructor which takes a pset with DeployProto and ModelFile fields. More...
int	NOutput () const
	Number of outputs in neural net. More...
int	NFeatures () const
	Number of outputs in neural net. More...
std::vector< std::vector< float > >	Predict (const PixelMap &pm)
	Return prediction arrays for PixelMap. More...
std::vector< float >	PredictFlavour (const PixelMap &pm)
	Return four element vector with summed numu, nue, nutau and NC elements. More...

Private Attributes
std::string	fLibPath
	Library path (typically dune_pardata...) More...
std::string	fTFProtoBuf
	location of the tf .pb file in the above path More...
bool	fUseLogChargeScale
	Is the charge using a log scale? More...
unsigned int	fImageWires
	Number of wires for the network to classify. More...
unsigned int	fImageTDCs
	Number of tdcs for the network to classify. More...
std::vector< bool >	fReverseViews
	Do we need to reverse any views? More...
std::unique_ptr< tf::Graph >	fTFGraph
	Tensorflow graph. More...

Detailed Description

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

Definition at line 22 of file TFNetHandler.h.

Constructor & Destructor Documentation

cvn::TFNetHandler::TFNetHandler

(

const fhicl::ParameterSet &

pset

)

Constructor which takes a pset with DeployProto and ModelFile fields.

Definition at line 22 of file TFNetHandler.cxx.

                                                         :
    fLibPath(cet::getenv(pset.get<std::string>("LibPath", ""))),
    fTFProtoBuf  (fLibPath+"/"+pset.get<std::string>("TFProtoBuf")),
    fUseLogChargeScale(pset.get<bool>("ChargeLogScale")),
    fImageWires(pset.get<unsigned int>("NImageWires")),
    fImageTDCs(pset.get<unsigned int>("NImageTDCs")),
    fReverseViews(pset.get<std::vector<bool> >("ReverseViews"))
  {

    // Construct the TF Graph object. The empty vector {} is used since the protobuf
    // file gives the names of the output layer nodes
    mf::LogInfo("TFNetHandler") << "Loading network: " << fTFProtoBuf << std::endl;
    fTFGraph = tf::Graph::create(fTFProtoBuf.c_str(),{},pset.get<int>("NInputs"),pset.get<int>("NOutputs"));
    if(!fTFGraph){
      art::Exception(art::errors::Unknown) << "Tensorflow model not found or incorrect";
    }

  }

Member Function Documentation

int cvn::TFNetHandler::NFeatures

(

)

const

Number of outputs in neural net.

int cvn::TFNetHandler::NOutput

(

)

const

Number of outputs in neural net.

std::vector< std::vector< float > > cvn::TFNetHandler::Predict

(

const PixelMap &

pm

)

Return prediction arrays for PixelMap.

Definition at line 69 of file TFNetHandler.cxx.

  {
   
    CVNImageUtils imageUtils;

    // Configure the image utility  
    imageUtils.SetViewReversal(fReverseViews);
    imageUtils.SetImageSize(fImageWires,fImageTDCs,3);
    imageUtils.SetLogScale(fUseLogChargeScale);

    ImageVectorF thisImage;
    imageUtils.ConvertPixelMapToImageVectorF(pm,thisImage);
    std::vector<ImageVectorF> vecForTF;

    vecForTF.push_back(thisImage);

    std::vector< std::vector< std::vector< float > > > cvnResults; // shape(samples, #outputs, output_size)
    bool status = false;

    int counter = 0;

    do{ // do until it gets a correct result
        // std::cout << "Number of CVN result vectors " << cvnResults.size() << " with " << cvnResults[0].size() << " categories" << std::endl;
        cvnResults = fTFGraph->run(vecForTF);
        status = check(cvnResults[0]);
        //std::cout << "Status: " << status << std::endl;
        counter++;
        if(counter==10){
            std::cout << "Error, CVN never outputing a correct result. Filling result with zeros.";
            std::cout << std::endl;
            fillEmpty(cvnResults[0]);
            break;
        }
    }while(status == false);

    std::cout << "Classifier summary: ";
    std::cout << std::endl;
    int output_index = 0;
    for(auto const & output : cvnResults[0])
    {
      std::cout << "Output " << output_index++ << ": ";
      for(auto const v : output)
          std::cout << v << ", ";
      std::cout << std::endl;
    }
    std::cout << std::endl;

    return cvnResults[0];
  }

std::vector<float> cvn::TFNetHandler::PredictFlavour

(

const PixelMap &

pm

)

Return four element vector with summed numu, nue, nutau and NC elements.

Member Data Documentation

unsigned int cvn::TFNetHandler::fImageTDCs

private

Number of tdcs for the network to classify.

Definition at line 47 of file TFNetHandler.h.

unsigned int cvn::TFNetHandler::fImageWires

private

Number of wires for the network to classify.

Definition at line 46 of file TFNetHandler.h.

std::string cvn::TFNetHandler::fLibPath

private

Library path (typically dune_pardata...)

Definition at line 43 of file TFNetHandler.h.

std::vector<bool> cvn::TFNetHandler::fReverseViews

private

Do we need to reverse any views?

Definition at line 48 of file TFNetHandler.h.

std::unique_ptr<tf::Graph> cvn::TFNetHandler::fTFGraph

private

Tensorflow graph.

Definition at line 49 of file TFNetHandler.h.

std::string cvn::TFNetHandler::fTFProtoBuf

private

location of the tf .pb file in the above path

Definition at line 44 of file TFNetHandler.h.

bool cvn::TFNetHandler::fUseLogChargeScale

private

Is the charge using a log scale?

Definition at line 45 of file TFNetHandler.h.

---

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

• dunereco/dunereco/CVN/art/TFNetHandler.h
• dunereco/dunereco/CVN/art/TFNetHandler.cxx