| | 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"] |
| | }, |
| | ) |
| |
|
