-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
34 lines (33 loc) · 969 Bytes
/
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
31
32
33
34
import os
from setuptools import setup, find_packages
setup(
name = "video_similarity_search",
version = "1.0",
packages=find_packages(),
install_requires=['numpy',
'scipy',
'pandas',
'matplotlib',
'scikit-learn',
'opencv-python',
'torch',
'torchvision',
'tensorboard',
'joblib',
'tqdm',
'ipdb',
'h5py',
'simplejson',
'fvcore',
'gspread',
'oauth2client',
'apiclient',
'google-api-python-client',
'torchviz'
], #external packages as dependencies
scripts=[
'train.py',
'evaluate.py',
'visualize.py'
]
)