Implementation of Pooling by Sliced-Wasserstein Embedding (NeurIPS 2021)
-
Updated
Nov 10, 2021 - Python
Implementation of Pooling by Sliced-Wasserstein Embedding (NeurIPS 2021)
Official implementation of the Convolutional Set Transformer (Chinello & Boracchi, 2025). This repository includes the source code of the cstmodels Python package, which provides reusable Keras 3 layers for constructing CST architectures, together with an interface to load and use the CST-15 model pre-trained on ImageNet.
Add a description, image, and links to the set-learning topic page so that developers can more easily learn about it.
To associate your repository with the set-learning topic, visit your repo's landing page and select "manage topics."