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