| import os |
|
|
| import setuptools |
|
|
|
|
| def read(rel_path): |
| base_path = os.path.abspath(os.path.dirname(__file__)) |
| with open(os.path.join(base_path, rel_path), "r") as f: |
| return f.read() |
|
|
|
|
| def get_version(rel_path): |
| for line in read(rel_path).splitlines(): |
| if line.startswith("__version__"): |
| |
| delim = '"' if '"' in line else "'" |
| return line.split(delim)[1] |
|
|
| raise RuntimeError("Unable to find version string.") |
|
|
|
|
| if __name__ == "__main__": |
| setuptools.setup( |
| name="pytorch-fid", |
| version=get_version(os.path.join("src", "pytorch_fid", "__init__.py")), |
| author="Max Seitzer", |
| description=( |
| "Package for calculating Frechet Inception Distance (FID)" " using PyTorch" |
| ), |
| long_description=read("README.md"), |
| long_description_content_type="text/markdown", |
| url="https://github.com/mseitzer/pytorch-fid", |
| package_dir={"": "src"}, |
| packages=setuptools.find_packages(where="src"), |
| classifiers=[ |
| "Programming Language :: Python :: 3", |
| "License :: OSI Approved :: Apache Software License", |
| ], |
| python_requires=">=3.5", |
| entry_points={ |
| "console_scripts": [ |
| "pytorch-fid = pytorch_fid.fid_score:main", |
| ], |
| }, |
| install_requires=[ |
| "numpy", |
| "pillow", |
| "scipy", |
| "torch>=1.0.1", |
| "torchvision>=0.2.2", |
| ], |
| extras_require={ |
| "dev": ["flake8", "flake8-bugbear", "flake8-isort", "black==24.3.0", "nox"] |
| }, |
| ) |
|
|
