File size: 858 Bytes
3176789 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | 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',
],
},
)
|