setup.py

import os

import setuptools

PACKAGE_NAME = "facenet-pytorch"
VERSION = "2.5.2"
AUTHOR = "Tim Esler"
EMAIL = "tim.esler@gmail.com"
DESCRIPTION = "Pretrained Pytorch face detection and recognition models"
GITHUB_URL = "https://github.com/timesler/facenet-pytorch"

parent_dir = os.path.dirname(os.path.realpath(__file__))
import_name = os.path.basename(parent_dir)

with open("{}/README.md".format(parent_dir), "r") as f:
    long_description = f.read()

setuptools.setup(
    name=PACKAGE_NAME,
    version=VERSION,
    author=AUTHOR,
    author_email=EMAIL,
    description=DESCRIPTION,
    long_description=long_description,
    long_description_content_type="text/markdown",
    url=GITHUB_URL,
    packages=["facenet_pytorch", "facenet_pytorch.models", "facenet_pytorch.models.utils", "facenet_pytorch.data"],
    package_data={"": ["*net*.pt"]},
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    install_requires=[
        "numpy>=1.24.0,<2.0.0",
        "Pillow>=10.2.0,<10.3.0",
        "requests>=2.0.0,<3.0.0",
        "torch>=2.2.0,<2.3.0",
        "torchvision>=0.17.0,<0.18.0",
        "tqdm>=4.0.0,<5.0.0",
    ],
)