Inheritance diagram for PointIdAlgTools::PointIdAlgTf:

Public Member Functions
	PointIdAlgTf (fhicl::Table< Config > const &table)

std::vector< float >	Run (std::vector< std::vector< float >> const &inp2d) const override

std::vector< std::vector< float > >	Run (std::vector< std::vector< std::vector< float >>> const &inps, int samples=-1) const override

Public Member Functions inherited from PointIdAlgTools::IPointIdAlg
virtual	~IPointIdAlg () noexcept=default

float	predictIdValue (unsigned int wire, float drift, size_t outIdx=0)

std::vector< float >	predictIdVector (unsigned int wire, float drift)

std::vector< std::vector< float > >	predictIdVectors (const std::vector< std::pair< unsigned int, float >> &points)

std::vector< std::string > const &	outputLabels (void) const

bool	isInsideFiducialRegion (unsigned int wire, float drift) const

Public Member Functions inherited from img::DataProviderAlg
	DataProviderAlg (const fhicl::ParameterSet &pset)

	DataProviderAlg (const Config &config)

virtual	~DataProviderAlg ()

bool	setWireDriftData (const detinfo::DetectorClocksData &clock_data, const detinfo::DetectorPropertiesData &det_prop, const std::vector< recob::Wire > &wires, unsigned int plane, unsigned int tpc, unsigned int cryo)

std::vector< float > const &	wireData (size_t widx) const

std::vector< std::vector< float > >	getPatch (size_t wire, float drift, size_t patchSizeW, size_t patchSizeD) const

float	getPixelOrZero (int wire, int drift) const

double	getAdcSum () const

size_t	getAdcArea () const

float	poolMax (int wire, int drift, size_t r=0) const
	Pool max value in a patch around the wire/drift pixel. More...

unsigned int	Cryo () const
	Pool sum of pixels in a patch around the wire/drift pixel. More...

unsigned int	TPC () const

unsigned int	Plane () const

unsigned int	NWires () const

unsigned int	NScaledDrifts () const

unsigned int	NCachedDrifts () const

unsigned int	DriftWindow () const

float	ZeroLevel () const
	Level of zero ADC after scaling. More...

double	LifetimeCorrection (detinfo::DetectorClocksData const &clock_data, detinfo::DetectorPropertiesData const &det_prop, double tick) const

Protected Member Functions
std::string	findFile (const char *fileName) const

Protected Member Functions inherited from PointIdAlgTools::IPointIdAlg
bool	bufferPatch (size_t wire, float drift, std::vector< std::vector< float >> &patch)

bool	bufferPatch (size_t wire, float drift)

void	resizePatch (void)

Protected Member Functions inherited from img::DataProviderAlg
std::vector< float >	downscaleMax (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const

std::vector< float >	downscaleMaxMean (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const

std::vector< float >	downscaleMean (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const

std::vector< float >	downscale (std::size_t dst_size, std::vector< float > const &adc, size_t tick0) const

size_t	getDriftIndex (float drift) const

std::optional< std::vector< float > >	setWireData (std::vector< float > const &adc, size_t wireIdx) const

bool	patchFromDownsampledView (size_t wire, float drift, size_t size_w, size_t size_d, std::vector< std::vector< float >> &patch) const

bool	patchFromOriginalView (size_t wire, float drift, size_t size_w, size_t size_d, std::vector< std::vector< float >> &patch) const

virtual DataProviderAlgView	resizeView (detinfo::DetectorClocksData const &clock_data, detinfo::DetectorPropertiesData const &det_prop, size_t wires, size_t drifts)

Private Attributes
std::unique_ptr< tf::Graph >	g

std::vector< std::string >	fNNetOutputPattern

std::string	fNNetModelFilePath

Additional Inherited Members
Public Types inherited from img::DataProviderAlg
enum	EDownscaleMode { kMax = 1, kMaxMean = 2, kMean = 3 }

Protected Attributes inherited from PointIdAlgTools::IPointIdAlg
std::vector< std::string >	fNNetOutputs

size_t	fPatchSizeW

size_t	fPatchSizeD

std::vector< std::vector< float > >	fWireDriftPatch

size_t	fCurrentWireIdx

size_t	fCurrentScaledDrift

Protected Attributes inherited from img::DataProviderAlg
DataProviderAlgView	fAlgView

EDownscaleMode	fDownscaleMode

size_t	fDriftWindow

bool	fDownscaleFullView

float	fDriftWindowInv

calo::CalorimetryAlg	fCalorimetryAlg

geo::GeometryCore const *	fGeometry

Detailed Description

Definition at line 20 of file PointIdAlgTf_tool.cc.

Constructor & Destructor Documentation

PointIdAlgTools::PointIdAlgTf::PointIdAlgTf ( fhicl::Table< Config > const & table )

explicit

Definition at line 38 of file PointIdAlgTf_tool.cc.

                                                             : img::DataProviderAlg(table())
   {
     // ... Get common config vars
     fNNetOutputs = table().NNetOutputs();
     fPatchSizeW = table().PatchSizeW();
     fPatchSizeD = table().PatchSizeD();
     fCurrentWireIdx = 99999;
     fCurrentScaledDrift = 99999;
 
     // ... Get "optional" config vars specific to tf interface
     std::string s_cfgvr;
     if (table().NNetModelFile(s_cfgvr)) { fNNetModelFilePath = s_cfgvr; }
     else {
       fNNetModelFilePath = "mycnn";
     }
     std::vector<std::string> vs_cfgvr;
     if (table().NNetOutputPattern(vs_cfgvr)) { fNNetOutputPattern = vs_cfgvr; }
     else {
       fNNetOutputPattern = {"cnn_output", "_netout"};
     }
 
     if ((fNNetModelFilePath.length() > 3) &&
         (fNNetModelFilePath.compare(fNNetModelFilePath.length() - 3, 3, ".pb") == 0)) {
       g = tf::Graph::create(findFile(fNNetModelFilePath.c_str()).c_str(), fNNetOutputPattern);
       if (!g) { throw art::Exception(art::errors::Unknown) << "TF model failed."; }
       mf::LogInfo("PointIdAlgTf") << "TF model loaded.";
     }
     else {
       mf::LogError("PointIdAlgTf") << "File name extension not supported.";
     }
 
     resizePatch();
   }

Member Function Documentation

std::string PointIdAlgTools::PointIdAlgTf::findFile ( const char * fileName ) const

protected

Definition at line 74 of file PointIdAlgTf_tool.cc.

   {
     std::string fname_out;
     cet::search_path sp("FW_SEARCH_PATH");
     if (!sp.find_file(fileName, fname_out)) {
       struct stat buffer;
       if (stat(fileName, &buffer) == 0) { fname_out = fileName; }
       else {
         throw art::Exception(art::errors::NotFound) << "Could not find the model file " << fileName;
       }
     }
     return fname_out;
   }

std::vector< float > PointIdAlgTools::PointIdAlgTf::Run ( std::vector< std::vector< float >> const & inp2d ) const

overridevirtual

Implements PointIdAlgTools::IPointIdAlg.

Definition at line 90 of file PointIdAlgTf_tool.cc.

   {
     long long int rows = inp2d.size(), cols = inp2d.front().size();
 
     tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, rows, cols, 1}));
     auto input_map = _x.tensor<float, 4>();
     for (long long int r = 0; r < rows; ++r) {
       const auto& row = inp2d[r];
       for (long long int c = 0; c < cols; ++c) {
         input_map(0, r, c, 0) = row[c];
       }
     }
 
     auto out = g->run(_x);
     if (!out.empty())
       return out.front();
     else
       return std::vector<float>();
   }

std::vector< std::vector< float > > PointIdAlgTools::PointIdAlgTf::Run	(	std::vector< std::vector< std::vector< float >>> const &	inps,
		int	samples = `-1`
	)		const

overridevirtual

Implements PointIdAlgTools::IPointIdAlg.

Definition at line 112 of file PointIdAlgTf_tool.cc.

   {
 
     if ((samples == 0) || inps.empty() || inps.front().empty() || inps.front().front().empty()) {
       return std::vector<std::vector<float>>();
     }
 
     if ((samples == -1) || (samples > (long long int)inps.size())) { samples = inps.size(); }
 
     long long int rows = inps.front().size(), cols = inps.front().front().size();
 
     tensorflow::Tensor _x(tensorflow::DT_FLOAT, tensorflow::TensorShape({samples, rows, cols, 1}));
     auto input_map = _x.tensor<float, 4>();
     for (long long int s = 0; s < samples; ++s) {
       const auto& sample = inps[s];
       for (long long int r = 0; r < rows; ++r) {
         const auto& row = sample[r];
         for (long long int c = 0; c < cols; ++c) {
           input_map(s, r, c, 0) = row[c];
         }
       }
     }
     return g->run(_x);
   }

Member Data Documentation

std::string PointIdAlgTools::PointIdAlgTf::fNNetModelFilePath

private

Definition at line 34 of file PointIdAlgTf_tool.cc.

std::vector<std::string> PointIdAlgTools::PointIdAlgTf::fNNetOutputPattern

private

Definition at line 33 of file PointIdAlgTf_tool.cc.

std::unique_ptr<tf::Graph> PointIdAlgTools::PointIdAlgTf::g

private

Definition at line 32 of file PointIdAlgTf_tool.cc.

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

larrecodnn/larrecodnn/ImagePatternAlgs/Tensorflow/Tools/PointIdAlgTf_tool.cc

Public Member Functions

Protected Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation