| from setuptools import setup, find_packages | |
| setup( | |
| name='facedet', | |
| version='1.0.0', | |
| description='Production Face Detection for Video — SCRFD + ByteTrack', | |
| author='FaceDet Team', | |
| python_requires='>=3.8', | |
| packages=find_packages(), | |
| install_requires=[ | |
| 'torch>=1.10', | |
| 'torchvision>=0.11', | |
| 'numpy>=1.20', | |
| 'opencv-python>=4.5', | |
| 'pyyaml>=5.4', | |
| ], | |
| extras_require={ | |
| 'deploy': ['onnx', 'onnxruntime-gpu', 'onnxsim'], | |
| 'eval': ['scipy', 'matplotlib'], | |
| 'dev': ['pytest', 'flake8', 'black'], | |
| }, | |
| entry_points={ | |
| 'console_scripts': [ | |
| 'facedet-train=scripts.train:main', | |
| 'facedet-eval=scripts.evaluate:main', | |
| 'facedet-video=scripts.detect_video:main', | |
| 'facedet-export=scripts.export:main', | |
| ], | |
| }, | |
| ) | |