Inheritance diagram for cnn::RegCNNPyTorch:

Public Member Functions
	RegCNNPyTorch (fhicl::ParameterSet const &pset)

	~RegCNNPyTorch ()

void	produce (art::Event &evt)

void	beginJob ()

void	endJob ()

Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)

template<typename Config >
	EDProducer (Table< Config > const &config)

std::string	workerType () const

Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept

	Producer (fhicl::ParameterSet const &)

	Producer (Producer const &)=delete

	Producer (Producer &&)=delete

Producer &	operator= (Producer const &)=delete

Producer &	operator= (Producer &&)=delete

void	doBeginJob (SharedResources const &resources)

void	doEndJob ()

void	doRespondToOpenInputFile (FileBlock const &fb)

void	doRespondToCloseInputFile (FileBlock const &fb)

void	doRespondToOpenOutputFiles (FileBlock const &fb)

void	doRespondToCloseOutputFiles (FileBlock const &fb)

bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)

bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)

bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)

bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)

bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)

Public Member Functions inherited from art::Modifier
	~Modifier () noexcept

	Modifier ()

	Modifier (Modifier const &)=delete

	Modifier (Modifier &&)=delete

Modifier &	operator= (Modifier const &)=delete

Modifier &	operator= (Modifier &&)=delete

Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept

	ModuleBase ()

ModuleDescription const &	moduleDescription () const

void	setModuleDescription (ModuleDescription const &)

std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const

void	sortConsumables (std::string const &current_process_name)

template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)

template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Attributes
std::string	fLibPath

std::string	fNetwork

std::string	fPixelMapInput

std::string	fResultLabel

torch::jit::script::Module	module

Additional Inherited Members
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer

using	WorkerType = WorkerT< EDProducer >

Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >

Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >

Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)

Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()

template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)

template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)

template<typename T , BranchType = InEvent>
void	consumesMany ()

template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)

template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)

template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Definition at line 27 of file RegCNNPyTorch_module.cc.

Constructor & Destructor Documentation

cnn::RegCNNPyTorch::RegCNNPyTorch ( fhicl::ParameterSet const & pset )

explicit

Definition at line 47 of file RegCNNPyTorch_module.cc.

                                                              :
         EDProducer(pset),
         fLibPath       (cet::getenv(pset.get<std::string>      ("LibPath", ""))),
         fNetwork       (fLibPath + "/" + pset.get<std::string> ("Network")),
         fPixelMapInput (pset.get<std::string>                  ("PixelMapInput")),
         fResultLabel   (pset.get<std::string>                  ("ResultLabel"))
     {
         produces<std::vector<cnn::RegCNNResult> >(fResultLabel);
     }

cnn::RegCNNPyTorch::~RegCNNPyTorch ( )

Definition at line 57 of file RegCNNPyTorch_module.cc.

                                   {
 
     }

Member Function Documentation

void cnn::RegCNNPyTorch::beginJob ( )

virtual

Reimplemented from art::EDProducer.

Definition at line 61 of file RegCNNPyTorch_module.cc.

                                  {
         std::cout<<"regcnn_torch job begins ...... "<<std::endl;
         try {
             // Deserialize the ScriptModule from a file using torch::jit::load().
             module = torch::jit::load(fNetwork);
         }
         catch (const c10::Error& e) {
             std::cerr<<"error loading the model\n";
             return;
         }
         mf::LogDebug("RegCNNPyTorch::beginJob")<<"loaded model "<<fNetwork<<" ... ok\n";
     }

void cnn::RegCNNPyTorch::endJob ( )

virtual

Reimplemented from art::EDProducer.

Definition at line 74 of file RegCNNPyTorch_module.cc.

74 {

75 }

void cnn::RegCNNPyTorch::produce ( art::Event & evt )

virtual

Define containers for the things we're going to produce

Load 3D pixel map for direction reco.

Implements art::EDProducer.

Definition at line 77 of file RegCNNPyTorch_module.cc.

                                              {
         /// Define containers for the things we're going to produce
         std::unique_ptr< std::vector<RegCNNResult> >
                                       resultCol(new std::vector<RegCNNResult>);
 
         /// Load 3D pixel map for direction reco.
         std::vector< art::Ptr< cnn::RegPixelMap3D > > pixelmap3Dlist;
         art::InputTag itag1(fPixelMapInput, fPixelMapInput);
         auto pixelmap3DListHandle = evt.getHandle< std::vector< cnn::RegPixelMap3D > >(itag1);
         if (pixelmap3DListHandle) {
             art::fill_ptr_vector(pixelmap3Dlist, pixelmap3DListHandle);
         }
 
         if (pixelmap3Dlist.size() > 0) {
             mf::LogDebug("RegCNNPyTorch::produce")<<"3D pixel map was made for this event, loading it as the input";
 
             RegPixelMap3D pm = *pixelmap3Dlist[0];
 
             // Convert RegPixelMap3D to at::Tensor, which is the actual input of the network
             // Currently we have two configurations for the 3D pixel map
             // 100*100*100: a pixel map centered at the vertex, need longer evaluation time
             // 32*32*32: cropped pixel map around the vertex of the interaction from 100*100*100 pm, faster
             at::Tensor t_pm;
             if (pm.IsCroppedPM()) {
                 t_pm = torch::from_blob(pm.fPECropped.data(), {1,1,32,32,32});
             } else {
                 t_pm = torch::from_blob(pm.fPE.data(), {1,1,100,100,100});
             }
 
             std::vector<torch::jit::IValue> inputs_pm;
             inputs_pm.push_back(t_pm);
             at::Tensor torchOutput = module.forward(inputs_pm).toTensor();
 
             // Output of the network
             // Currently only direction reconstruction utilizes 3D CNN, which has 3 output represent 3 components
             // Absolute value of 3 output are meaningless, their combination is the direction of the prong
             std::vector<float> networkOutput;
             for (unsigned int i= 0; i< 3; ++i) {
                 networkOutput.push_back(torchOutput[0][i].item<float>());
                 std::cout<<torchOutput[0][i].item<float>()<<std::endl;
             }
 
             resultCol->emplace_back(networkOutput);
 
             //std::cout<<output.slice(/*dim=*/1, /*start=*/0, /*end=*/10) << '\n';
             //std::cout<<output[0]<<std::endl;
         }
 
         evt.put(std::move(resultCol), fResultLabel);
     }

Member Data Documentation

std::string cnn::RegCNNPyTorch::fLibPath

private

Definition at line 39 of file RegCNNPyTorch_module.cc.

std::string cnn::RegCNNPyTorch::fNetwork

private

Definition at line 40 of file RegCNNPyTorch_module.cc.

std::string cnn::RegCNNPyTorch::fPixelMapInput

private

Definition at line 41 of file RegCNNPyTorch_module.cc.

std::string cnn::RegCNNPyTorch::fResultLabel

private

Definition at line 42 of file RegCNNPyTorch_module.cc.

torch::jit::script::Module cnn::RegCNNPyTorch::module

private

Definition at line 44 of file RegCNNPyTorch_module.cc.

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

dunereco/dunereco/RegCNN/art/RegCNNPyTorch_module.cc

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation