- C++ with ISO 14 standard (GCC 6.1 and above)
- CMake >= 3.0
- CUDA >= 9.1 (highly recommended because of the speed-up)
- conan.io (optional for C++ dependencies) or
- PyBind11 (optional for Python interface)
- google-test 1.8.1 (optional for unit tests)
- doxygen 1.8.13 (optional for developer documentation)
Conan.io will install automatically the C++ dependencies (PyBind11 and google-test). Otherwise you can also install these libraries yourself.
We provide deb- and rpm-packages at https://github.com/HITS-AIN/PINK/releases
or you can install PINK from the sources:
cmake -DCMAKE_INSTALL_PREFIX=<INSTALL_PATH> .
make install
To train a the self-organizing map (SOM) please execute
Pink --train <image-file> <result-file>
where image-file
is the input file of images for the training and result-file
is the output file for the trained SOM. All files are in binary mode described here.
To map an image to the trained SOM please execute
Pink --map <image-file> <result-file> <SOM-file>
where image-file
is the input file of images for the mapping, SOM-file
is the input file for the trained SOM, and result-file
is the output file for the resulting heatmap.
Please use also the command Pink -h
to get more informations about the usage and the options.
For conversion and visualization of images and SOM some python scripts are available.
- convert_data_binary_file.py Convert binary data file from PINK version 1 to 2
- data_precondition.py: Convert numpy into binary files and scale data to be within the range [0.0, 1.0]
- show_heatmap.py: Visualize the mapping result
- show_images.py: Visualize binary images file format
- show_som.py: Visualize binary SOM file format
- train.py: SOM training using the PINK Python interface
The input data for the SOM training are radio-synthesis images of Radio Galaxy Zoo containing 176750 images of the dimension 124x124. The SOM layout is hexagonal of the dimension 21x21 which has 331 neurons (see image above). The size of the neurons is 64x64. The accuracy for the rotational invariance is 1 degree and the flip invariance is used.
CPU: 2x Intel Gold 5118 (12 cores in total)
GPU: 4x NVIDIA Tesla P40
PINK 1.1 | Pink 2.0 | |
---|---|---|
CPU | ||
CPU + 1x GPU | 3069 | 909 |
CPU + 2x GPU | 2069 | 636 |
CPU + 4x GPU | 1891 | 858 |
Kai Lars Polsterer, Fabian Gieseke, Christian Igel, Bernd Doser, and Nikos Gianniotis. Parallelized rotation and flipping INvariant Kohonen maps (PINK) on GPUs. 24th European Symposium on Artificial Neural Networks, Computational Intelligence and Machine Learning (ESANN), pp. 405-410, 2016. pdf
Distributed under the GNU GPLv3 License. See accompanying file LICENSE or copy at http://www.gnu.org/licenses/gpl-3.0.html.