forked from aleju/imgaug
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
37 lines (35 loc) · 1.32 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from setuptools import setup, find_packages
long_description = """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."""
setup(
name="imgaug",
version="0.3.0",
author="Alexander Jung",
author_email="[email protected]",
url="https://github.com/aleju/imgaug",
download_url="https://github.com/aleju/imgaug/archive/0.3.0.tar.gz",
install_requires=[
"six",
"numpy>=1.15",
"scipy",
"Pillow",
"matplotlib",
"scikit-image>=0.14.2",
"opencv-python-headless",
"imageio",
"Shapely",
],
packages=find_packages(),
include_package_data=True,
package_data={
"": ["LICENSE", "README.md", "requirements.txt"],
"imgaug": ["DejaVuSans.ttf", "quokka.jpg", "quokka_annotations.json", "quokka_depth_map_halfres.png"],
"imgaug.checks": ["README.md"]
},
license="MIT",
description="Image augmentation library for deep neural networks",
long_description=long_description,
keywords=["augmentation", "image", "deep learning", "neural network", "CNN", "machine learning",
"computer vision", "overfitting"]
)