I am trying to use the CNN in TMVA to analys pixelized electron track images.
The images are 3D voxels of the track, with high resolution in XY and coarse resolution in Z. So I am treating the Z direction as the color channels in CNN. The images are 16x16x3 pixels in x,y and z.
The example https://root.cern/doc/master/TMVA__CNN__Classification_8C.html mentioned the possiblity to have multi-channel images as input with the
AddVariablesArray() methode, and the inputLayout is
For a 3 channel image, I modify the layout to
3|16|16 , but how should I order the pixels in the array? Should it be
vector<int> var with index
i_z*256 + i_y*16 +i_x or
i_y*48 + i_x*3 +i_z ?
Also I didn’t find the corresponding configuration for the layer layouts. Can I keep the same batch layout as in the tutorial example?
Thank you very much!