-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
30 lines (28 loc) · 1.13 KB
/
setup.py
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
"""setup script for the PyVideoAI package."""
from setuptools import setup
import versioneer
setup(
name = "PyVideoAI",
setup_requires=[],
version=versioneer.get_version(),
cmdclass=versioneer.get_cmdclass(),
author = "Kiyoon Kim",
author_email='kiyoon.kim@ed.ac.uk',
description = "Video datasets' annotation parser and etc.",
url = "https://github.com/kiyoon/PyVideoAI",
packages=['pyvideoai', 'dataset_configs', 'model_configs', 'exp_configs'],
python_requires='>=3.6',
install_requires=['numpy>=1.16.0',
'coloredlogs', 'verboselogs',
'matplotlib', 'scipy', 'scikit-learn',
'seaborn', # confusion matrix plots
'pandas>=1.3.1', # pandas 1.3.0 has a bug when loading pickled data
'tensorboard',
'opencv-python',
'moviepy', # Tensorboard `add_video()`
'decord', # Video decoding
# 'av', # Video decoding (you may not need it in favour of decord)
'scikit-image', # Only used for timecycle
'gulpio2', # Efficient frame dataloader
],
)