|
|
from setuptools import setup, find_packages |
|
|
|
|
|
with open("README.md", "r", encoding="utf-8") as fh: |
|
|
long_description = fh.read() |
|
|
|
|
|
setup( |
|
|
name="imgshape", |
|
|
version="2.1.3", |
|
|
description="Smart image shape, dataset analysis, preprocessing & augmentation recommendations for ML workflows", |
|
|
long_description=long_description, |
|
|
long_description_content_type="text/markdown", |
|
|
author="Stifler", |
|
|
author_email="stiflerxd.ai@cudabit.live", |
|
|
packages=find_packages(where="src"), |
|
|
package_dir={"": "src"}, |
|
|
install_requires=[ |
|
|
"Pillow>=9.0.0", |
|
|
"numpy>=1.21.0", |
|
|
"matplotlib>=3.4.0", |
|
|
"scikit-image>=0.19.0", |
|
|
"gradio>=3.0.0", |
|
|
|
|
|
"seaborn>=0.11.0", |
|
|
], |
|
|
extras_require={ |
|
|
"torch": ["torch>=1.12.0", "torchvision>=0.13.0"], |
|
|
"pdf": ["weasyprint>=53.0"], |
|
|
"dev": ["pytest>=7.0", "black>=23.0", "flake8>=3.9", "pre-commit>=2.20"], |
|
|
}, |
|
|
entry_points={ |
|
|
"console_scripts": [ |
|
|
"imgshape=imgshape.cli:main", |
|
|
] |
|
|
}, |
|
|
classifiers=[ |
|
|
"Programming Language :: Python :: 3", |
|
|
"License :: OSI Approved :: MIT License", |
|
|
"Operating System :: OS Independent", |
|
|
"Topic :: Scientific/Engineering :: Image Recognition", |
|
|
], |
|
|
python_requires=">=3.8", |
|
|
keywords="images dataset visualization augmentation pytorch ml", |
|
|
url="https://github.com/STiFLeR7/imgshape", |
|
|
project_urls={ |
|
|
"Source": "https://github.com/STiFLeR7/imgshape", |
|
|
"Issues": "https://github.com/STiFLeR7/imgshape/issues", |
|
|
}, |
|
|
) |
|
|
|
