![]() |
LArSoft
v10_04_05
Liquid Argon Software toolkit - https://larsoft.org/
|
DeepLearningTrackShowerIdAlgorithm class. More...
#include "DlVertexingBaseAlgorithm.h"
Public Types | |
typedef std::map< std::pair< int, int >, std::vector< const pandora::CaloHit * > > | PixelToCaloHitsMap |
Public Member Functions | |
DlVertexingBaseAlgorithm () | |
Default constructor. More... | |
~DlVertexingBaseAlgorithm () | |
Protected Types | |
typedef std::pair< int, int > | Pixel |
typedef std::vector< Pixel > | PixelVector |
Protected Member Functions | |
pandora::StatusCode | ReadSettings (const pandora::TiXmlHandle xmlHandle) |
void | GetHitRegion (const pandora::CaloHitList &caloHitList, float &xMin, float &xMax, float &zMin, float &zMax) const |
void | GetCanvasParameters (const LArDLHelper::TorchOutput &networkOutput, const PixelVector &pixelVector, int &columnOffset, int &rowOffset, int &width, int &height) const |
Determines the parameters of the canvas for extracting the vertex location. The network predicts the distance that each pixel associated with a hit is located from the vertex, but says nothing about the direction. As a result, the ring describing the potential vertices associated with that hit can extend beyond the original canvas size. This function returns the size of the required canvas and the offset for the bottom left corner. More... | |
Protected Attributes | |
bool | m_trainingMode |
Training mode. More... | |
std::string | m_trainingOutputFile |
Output file name for training examples. More... | |
std::string | m_inputVertexListName |
Input vertex list name if 2nd pass. More... | |
std::string | m_outputVertexListName |
Output vertex list name. More... | |
pandora::StringVector | m_caloHitListNames |
Names of input calo hit lists. More... | |
LArDLHelper::TorchModel | m_modelU |
The model for the U view. More... | |
LArDLHelper::TorchModel | m_modelV |
The model for the V view. More... | |
LArDLHelper::TorchModel | m_modelW |
The model for the W view. More... | |
int | m_pass |
The pass of the train/infer step. More... | |
int | m_nClasses |
The number of distance classes. More... | |
int | m_height |
The height of the images. More... | |
int | m_width |
The width of the images. More... | |
float | m_driftStep |
The size of a pixel in the drift direction in cm (most relevant in pass 2) More... | |
std::vector< double > | m_thresholds |
Distance class thresholds. More... | |
std::string | m_volumeType |
The name of the fiducial volume type for the monitoring output. More... | |
DeepLearningTrackShowerIdAlgorithm class.
Definition at line 28 of file DlVertexingBaseAlgorithm.h.
|
protected |
Definition at line 41 of file DlVertexingBaseAlgorithm.h.
typedef std::map<std::pair<int, int>, std::vector<const pandora::CaloHit *> > lar_dl_content::DlVertexingBaseAlgorithm::PixelToCaloHitsMap |
Definition at line 31 of file DlVertexingBaseAlgorithm.h.
|
protected |
Definition at line 42 of file DlVertexingBaseAlgorithm.h.
lar_dl_content::DlVertexingBaseAlgorithm::DlVertexingBaseAlgorithm | ( | ) |
Default constructor.
Definition at line 30 of file DlVertexingBaseAlgorithm.cc.
References m_driftStep, m_height, m_nClasses, m_pass, m_trainingOutputFile, m_volumeType, and m_width.
lar_dl_content::DlVertexingBaseAlgorithm::~DlVertexingBaseAlgorithm | ( | ) |
Definition at line 44 of file DlVertexingBaseAlgorithm.cc.
|
protected |
Determines the parameters of the canvas for extracting the vertex location. The network predicts the distance that each pixel associated with a hit is located from the vertex, but says nothing about the direction. As a result, the ring describing the potential vertices associated with that hit can extend beyond the original canvas size. This function returns the size of the required canvas and the offset for the bottom left corner.
networkOutput | The TorchOutput object populated by the network inference step |
pixelVector | The vector of populated pixels |
columnOffset | The output column offset for the canvas |
rowOffset | The output row offset for the canvas |
width | The output width for the canvas |
height | The output height for the canvas |
Definition at line 165 of file DlVertexingBaseAlgorithm.cc.
References col, m_height, m_thresholds, and m_width.
Referenced by lar_dl_content::DlVertexingAlgorithm::Infer(), and lar_dl_content::DlSecondaryVertexingAlgorithm::Infer().
|
protected |
Definition at line 50 of file DlVertexingBaseAlgorithm.cc.
References util::begin(), m_caloHitListNames, m_driftStep, m_height, m_inputVertexListName, m_pass, m_width, x, and z.
Referenced by lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), lar_dl_content::DlVertexingAlgorithm::PrepareTrainingSample(), and lar_dl_content::DlSecondaryVertexingAlgorithm::PrepareTrainingSample().
|
protected |
Definition at line 200 of file DlVertexingBaseAlgorithm.cc.
References lar_dl_content::LArDLHelper::LoadModel(), m_caloHitListNames, m_driftStep, m_height, m_inputVertexListName, m_modelU, m_modelV, m_modelW, m_nClasses, m_outputVertexListName, m_pass, m_thresholds, m_trainingMode, m_trainingOutputFile, m_volumeType, and m_width.
Referenced by lar_dl_content::DlVertexingAlgorithm::ReadSettings(), and lar_dl_content::DlSecondaryVertexingAlgorithm::ReadSettings().
|
protected |
Names of input calo hit lists.
Definition at line 79 of file DlVertexingBaseAlgorithm.h.
Referenced by GetHitRegion(), lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), lar_dl_content::DlVertexingAlgorithm::PrepareTrainingSample(), lar_dl_content::DlSecondaryVertexingAlgorithm::PrepareTrainingSample(), and ReadSettings().
|
protected |
The size of a pixel in the drift direction in cm (most relevant in pass 2)
Definition at line 87 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), GetHitRegion(), lar_dl_content::DlVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlVertexingAlgorithm::MakeWirePlaneCoordinatesFromCanvas(), and ReadSettings().
|
protected |
The height of the images.
Definition at line 85 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), GetCanvasParameters(), GetHitRegion(), lar_dl_content::DlSecondaryVertexingAlgorithm::GetVerticesFromCanvas(), lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), lar_dl_content::DlVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlSecondaryVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlVertexingAlgorithm::MakeWirePlaneCoordinatesFromCanvas(), and ReadSettings().
|
protected |
Input vertex list name if 2nd pass.
Definition at line 77 of file DlVertexingBaseAlgorithm.h.
Referenced by GetHitRegion(), lar_dl_content::DlVertexingAlgorithm::Infer(), and ReadSettings().
|
protected |
The model for the U view.
Definition at line 80 of file DlVertexingBaseAlgorithm.h.
Referenced by lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), and ReadSettings().
|
protected |
The model for the V view.
Definition at line 81 of file DlVertexingBaseAlgorithm.h.
Referenced by lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), and ReadSettings().
|
protected |
The model for the W view.
Definition at line 82 of file DlVertexingBaseAlgorithm.h.
Referenced by lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), and ReadSettings().
|
protected |
The number of distance classes.
Definition at line 84 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), and ReadSettings().
|
protected |
Output vertex list name.
Definition at line 78 of file DlVertexingBaseAlgorithm.h.
Referenced by lar_dl_content::DlVertexingAlgorithm::MakeCandidateVertexList(), lar_dl_content::DlSecondaryVertexingAlgorithm::MakeCandidateVertexList(), and ReadSettings().
|
protected |
The pass of the train/infer step.
Definition at line 83 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), GetHitRegion(), lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlVertexingAlgorithm::MakeCandidateVertexList(), lar_dl_content::DlVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlSecondaryVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlVertexingAlgorithm::PrepareTrainingSample(), and ReadSettings().
|
protected |
Distance class thresholds.
Definition at line 88 of file DlVertexingBaseAlgorithm.h.
Referenced by GetCanvasParameters(), lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), and ReadSettings().
|
protected |
Training mode.
Definition at line 75 of file DlVertexingBaseAlgorithm.h.
Referenced by lar_dl_content::DlVertexingAlgorithm::ReadSettings(), ReadSettings(), lar_dl_content::DlSecondaryVertexingAlgorithm::ReadSettings(), lar_dl_content::DlVertexingAlgorithm::Run(), and lar_dl_content::DlSecondaryVertexingAlgorithm::Run().
|
protected |
Output file name for training examples.
Definition at line 76 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), lar_dl_content::DlVertexingAlgorithm::PrepareTrainingSample(), lar_dl_content::DlSecondaryVertexingAlgorithm::PrepareTrainingSample(), and ReadSettings().
|
protected |
The name of the fiducial volume type for the monitoring output.
Definition at line 89 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), lar_dl_content::DlVertexingAlgorithm::MakeCandidateVertexList(), lar_dl_content::DlSecondaryVertexingAlgorithm::PrepareTrainingSample(), and ReadSettings().
|
protected |
The width of the images.
Definition at line 86 of file DlVertexingBaseAlgorithm.h.
Referenced by DlVertexingBaseAlgorithm(), GetCanvasParameters(), GetHitRegion(), lar_dl_content::DlSecondaryVertexingAlgorithm::GetVerticesFromCanvas(), lar_dl_content::DlVertexingAlgorithm::Infer(), lar_dl_content::DlSecondaryVertexingAlgorithm::Infer(), lar_dl_content::DlVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlSecondaryVertexingAlgorithm::MakeNetworkInputFromHits(), lar_dl_content::DlVertexingAlgorithm::MakeWirePlaneCoordinatesFromCanvas(), and ReadSettings().