summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-imgaug/package.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-imgaug/package.py b/var/spack/repos/builtin/packages/py-imgaug/package.py
new file mode 100644
index 0000000000..461f17bdd2
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-imgaug/package.py
@@ -0,0 +1,31 @@
+# Copyright 2013-2021 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.999', type=('build', 'run'))
+ depends_on('opencv+python3', when='^python@3:', type=('build', 'run'))