spack/var/spack/repos/builtin/packages/py-imgaug/package.py
2022-01-14 22:50:21 -08:00

32 lines
1.4 KiB
Python

# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from spack import *
class PyImgaug(PythonPackage):
"""A library for image augmentation in machine learning experiments,
particularly convolutional neural networks. Supports the augmentation of
images, keypoints/landmarks, bounding boxes, heatmaps and segmentation maps
in a variety of different ways."""
homepage = "https://github.com/aleju/imgaug"
pypi = "imgaug/imgaug-0.3.0.tar.gz"
version('0.4.0', sha256='46bab63ed38f8980630ff721a09ca2281b7dbd4d8c11258818b6ebcc69ea46c7')
version('0.3.0', sha256='e1354d41921f1b306b50c5141b4870f17e81b531cae2f5c3093da9dc4dcb3cf4')
depends_on('py-setuptools', type='build')
depends_on('py-imageio', type=('build', 'run'))
depends_on('py-matplotlib', type=('build', 'run'))
depends_on('py-numpy@1.15:', type=('build', 'run'))
depends_on('py-scipy', type=('build', 'run'))
depends_on('py-shapely', type=('build', 'run'))
depends_on('py-six', type=('build', 'run'))
depends_on('py-scikit-image@0.14.2:', type=('build', 'run'))
depends_on('pil', type=('build', 'run'))
depends_on('opencv+python2', when='^python@2.0:2', type=('build', 'run'))
depends_on('opencv+python3', when='^python@3:', type=('build', 'run'))