object-condensation 1.0.0

Efficient implementations of the Object Condensation losses (Jan Kieseler, 2020)

object_condensation

The Object Condensation loss - developed by Jan Kieseler - is now being used by several groups in high energy physics for both track reconstruction and shower reconstruction in calorimeters.

Several implementations of this idea already exist, but often they are maintained by very few people. This repository aims to provide an easy to use implementation for both the TensorFlow and PyTorch backend.

Existing Implementations:

• cms-pepr [TensorFlow]
• mlpf [PyTorch]
• gnn-tracking [PyTorch]

Installation

Clone project and run

python3 -m pip install -e '.[pytorch]'
# or
python3 -m pip install -e '.[tensorflow]'

Development setup

For the development setup, also add dev and testing, e.g.,

python3 -m pip install -e '.[pytorch,dev,testing]'

Please also install pre-commit:

python3 -m pip install pre-commit
pre-commit install  # in top-level directory of repository