-
Notifications
You must be signed in to change notification settings - Fork 178
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: Adding a few changes to test seeding algorithm (#4075)
This PR includes a few changes to facilitate performance studies for the seeding algorithm @pbutti, @benjaminhuth <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **New Features** - Enhanced seeding capability with new configuration options for additional filtering and customizable seed naming. - Introduced a demonstration script that simplifies data reading and seeding, now featuring improved debug logging. - **Chores** - Optimized tracking criteria by refining selection thresholds and updating filtering parameters for improved performance. - Added debug logging to track space point creation for better visibility during processing. - Introduced a new parameter for occupancy configuration in the seeding algorithm. - Updated filtering criteria in the reader to focus on space points. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
- Loading branch information
Showing
6 changed files
with
69 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,59 @@ | ||
import sys | ||
from pathlib import Path | ||
|
||
import acts.examples | ||
import acts | ||
|
||
from acts.examples import ( | ||
CsvSpacePointReader, | ||
TrackParamsEstimationAlgorithm, | ||
SeedingPerformanceWriter, | ||
) | ||
from acts.examples.reconstruction import ( | ||
addStandardSeeding, | ||
) | ||
|
||
from acts.examples.itk import itkSeedingAlgConfig, InputSpacePointsType | ||
|
||
|
||
s = acts.examples.Sequencer(events=1, numThreads=1, outputDir="output") | ||
|
||
# loggingLevel = acts.logging.INFO | ||
loggingLevel = acts.logging.DEBUG | ||
|
||
s.addReader( | ||
acts.examples.RootAthenaDumpReader( | ||
level=loggingLevel, | ||
treename="GNN4ITk", | ||
inputfile="Dump_GNN4Itk.root", | ||
onlySpacepoints=True, | ||
outputPixelSpacePoints="pixel_spacepoints", | ||
outputStripSpacePoints="strip_spacepoints", | ||
outputSpacePoints="spacepoints", | ||
) | ||
) | ||
|
||
|
||
# run pixel seeding | ||
seeding_pixel = addStandardSeeding( | ||
s, | ||
"pixel_spacepoints", | ||
*acts.examples.itk.itkSeedingAlgConfig( | ||
InputSpacePointsType.PixelSpacePoints, highOccupancyConfig=True | ||
), | ||
logLevel=loggingLevel, | ||
outputSeeds="pixel_seeds" | ||
) | ||
|
||
|
||
# run strip seeding | ||
seeding_strip = addStandardSeeding( | ||
s, | ||
"strip_spacepoints", | ||
*acts.examples.itk.itkSeedingAlgConfig(InputSpacePointsType.StripSpacePoints), | ||
logLevel=acts.logging.DEBUG, | ||
outputSeeds="strip_seeds" | ||
) | ||
|
||
|
||
s.run() |