File size: 2,473 Bytes
059f915 | 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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | """
VintageGAN Package Setup
Installation script for VintageGAN project.
Author: VintageGAN Project
Date: 2024
"""
from setuptools import setup, find_packages
from pathlib import Path
# Read README for long description
readme_file = Path(__file__).parent / "README.md"
long_description = (
readme_file.read_text(encoding="utf-8") if readme_file.exists() else ""
)
# Read requirements
requirements_file = Path(__file__).parent / "requirements.txt"
requirements = []
if requirements_file.exists():
with open(requirements_file, "r") as f:
requirements = [
line.strip() for line in f if line.strip() and not line.startswith("#")
]
setup(
name="vintagegan",
version="0.2.0",
author="VintageGAN Project",
author_email="",
description="Controllable vintage image synthesis with conditional GANs",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/msrishav-28/vintage-gan",
packages=find_packages(
exclude=["tests", "notebooks", "data", "checkpoints", "logs", "outputs"]
),
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"Intended Audience :: Developers",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Image Processing",
],
python_requires=">=3.9",
install_requires=requirements,
extras_require={
"dev": [
"pytest>=7.0.0",
"black>=22.0.0",
"flake8>=4.0.0",
"mypy>=0.950",
],
"notebook": [
"jupyter>=1.0.0",
"ipywidgets>=7.6.0",
],
},
entry_points={
"console_scripts": [
"vintagegan-filter=inference.apply_filter:main",
"vintagegan-batch=inference.batch_process:main",
"vintagegan-pretrain=training.pretrain:main",
"vintagegan-train=training.gan_train:main",
"vintagegan-ablation=evaluation.ablation:main",
],
},
include_package_data=True,
zip_safe=False,
)
|