numpy
torch
torchvision
packaging

[segment]
pgpd>=2
brambox>=3.3

[training]
brambox>=2
scikit-learn
