Inheritance diagram for SemanticDecoder:

Public Member Functions
	SemanticDecoder (const fhicl::ParameterSet &pset)
	Constructor. More...

virtual	~SemanticDecoder () noexcept=default
	Virtual Destructor. More...

void	declareProducts (art::ProducesCollector &collector) override
	declareProducts function More...

void	writeEmptyToEvent (art::Event &e, const vector< vector< size_t >> &idsmap) override
	writeEmptyToEvent function More...

void	writeToEvent (art::Event &e, const vector< vector< size_t >> &idsmap, const vector< NuGraphOutput > &infer_output) override
	Decoder function. More...

void	printVector (const std::vector< float > &vec)

template<typename T , size_t N>
void	softmax (std::array< T, N > &arr)

void	setDebugAndPlanes (bool d, vector< std::string > &p)

Protected Attributes
bool	debug

vector< std::string >	planes

std::string	instancename

std::string	outputname

Private Attributes
std::vector< std::string >	categories

art::InputTag	hitInput

Detailed Description

Definition at line 12 of file SemanticDecoder_tool.cc.

Constructor & Destructor Documentation

SemanticDecoder::SemanticDecoder ( const fhicl::ParameterSet & pset )

Constructor.

Parameters

pset

Definition at line 59 of file SemanticDecoder_tool.cc.

References fhicl::ParameterSet::get(), and hitInput.

   : DecoderToolBase(p)
   , categories{p.get<std::vector<std::string>>("categories")}
   , hitInput{p.get<art::InputTag>("hitInput")}
 {}

virtual SemanticDecoder::~SemanticDecoder ( )

virtualdefaultnoexcept

Virtual Destructor.

Member Function Documentation

void SemanticDecoder::declareProducts ( art::ProducesCollector & collector )

inlineoverridevirtual

declareProducts function

Parameters

art::ProducesCollector

Implements DecoderToolBase.

Definition at line 32 of file SemanticDecoder_tool.cc.

References e, DecoderToolBase::instancename, art::ProducesCollector::produces(), lar::dump::vector(), writeEmptyToEvent(), and writeToEvent().

   {
     collector.produces<vector<FeatureVector<5>>>(instancename);
     collector.produces<MVADescription<5>>(instancename);
   }

void DecoderToolBase::printVector ( const std::vector< float > & vec )

inlineinherited

Definition at line 65 of file DecoderToolBase.h.

Referenced by FilterDecoder::writeToEvent(), and writeToEvent().

   {
     for (size_t i = 0; i < vec.size(); ++i) {
       std::cout << vec[i];
       // Print space unless it's the last element
       if (i != vec.size() - 1) { std::cout << " "; }
     }
     std::cout << std::endl;
     std::cout << std::endl;
   }

void DecoderToolBase::setDebugAndPlanes	(	bool	d,
		vector< std::string > &	p
	)

inlineinherited

Definition at line 94 of file DecoderToolBase.h.

References d, and tca::debug.

   {
     debug = d;
     planes = p;
   }

template<typename T , size_t N>

void DecoderToolBase::softmax ( std::array< T, N > & arr )

inlineinherited

Definition at line 77 of file DecoderToolBase.h.

References sum.

Referenced by writeToEvent().

   {
     T m = -std::numeric_limits<T>::max();
     for (size_t i = 0; i < arr.size(); i++) {
       if (arr[i] > m) { m = arr[i]; }
     }
     T sum = 0.0;
     for (size_t i = 0; i < arr.size(); i++) {
       sum += expf(arr[i] - m);
     }
     T offset = m + logf(sum);
     for (size_t i = 0; i < arr.size(); i++) {
       arr[i] = expf(arr[i] - offset);
     }
     return;
   }

void SemanticDecoder::writeEmptyToEvent	(	art::Event &	e,
		const vector< vector< size_t >> &	idsmap
	)

overridevirtual

writeEmptyToEvent function

Parameters

art::Event event record

Implements DecoderToolBase.

Definition at line 65 of file SemanticDecoder_tool.cc.

References categories, trkf::fill(), hitInput, DecoderToolBase::instancename, art::InputTag::label(), art::Event::put(), and util::size().

Referenced by declareProducts().

 {
   //
   auto semtdes = std::make_unique<MVADescription<5>>(hitInput.label(), instancename, categories);
   e.put(std::move(semtdes), instancename);
   //
   size_t size = 0;
   for (auto& v : idsmap)
     size += v.size();
   std::array<float, 5> arr;
   std::fill(arr.begin(), arr.end(), -1.);
   auto semtcol = std::make_unique<vector<FeatureVector<5>>>(size, FeatureVector<5>(arr));
   e.put(std::move(semtcol), instancename);
   //
 }

void SemanticDecoder::writeToEvent	(	art::Event &	e,
		const vector< vector< size_t >> &	idsmap,
		const vector< NuGraphOutput > &	infer_output
	)

overridevirtual

Decoder function.

Parameters

art::Event event record for decoder

Implements DecoderToolBase.

Definition at line 81 of file SemanticDecoder_tool.cc.

References categories, DecoderToolBase::debug, DEFINE_ART_CLASS_TOOL, trkf::fill(), hitInput, DecoderToolBase::instancename, art::InputTag::label(), DecoderToolBase::outputname, DecoderToolBase::planes, DecoderToolBase::printVector(), art::Event::put(), util::size(), and DecoderToolBase::softmax().

Referenced by declareProducts().

 {
   //
   auto semtdes = std::make_unique<MVADescription<5>>(hitInput.label(), instancename, categories);
   e.put(std::move(semtdes), instancename);
   //
   size_t size = 0;
   for (auto& v : idsmap)
     size += v.size();
   std::array<float, 5> arr;
   std::fill(arr.begin(), arr.end(), -1.);
   auto semtcol = std::make_unique<vector<FeatureVector<5>>>(size, FeatureVector<5>(arr));
 
   size_t n_cols = categories.size();
   for (size_t p = 0; p < planes.size(); p++) {
     //
     const std::vector<float>* x_semantic_data = 0;
     for (auto& io : infer_output) {
       if (io.output_name == outputname + planes[p]) x_semantic_data = &io.output_vec;
     }
     if (debug) {
       std::cout << outputname + planes[p] << std::endl;
       printVector(*x_semantic_data);
     }
 
     torch::TensorOptions options = torch::TensorOptions().dtype(torch::kFloat32);
     size_t n_rows = x_semantic_data->size() / n_cols;
     const torch::Tensor s =
       torch::from_blob(const_cast<float*>(x_semantic_data->data()),
                        {static_cast<int64_t>(n_rows), static_cast<int64_t>(n_cols)},
                        options);
 
     for (int i = 0; i < s.sizes()[0]; ++i) {
       size_t idx = idsmap[p][i];
       std::array<float, 5> input;
       for (size_t j = 0; j < n_cols; ++j)
         input[j] = s[i][j].item<float>();
       softmax(input);
       FeatureVector<5> semt = FeatureVector<5>(input);
       (*semtcol)[idx] = semt;
     }
   }
   e.put(std::move(semtcol), instancename);
 }