VideoSuperResolution/setup.py at master · zhjuly/VideoSuperResolution · GitHub

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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
"""
Copyright: Wenyi Tang 2017-2019
Author: Wenyi Tang
Email: wenyi.tang@intel.com
Last update date: Mar. 25th 2019
"""

#  Copyright (c): Wenyi Tang 2017-2019.
#  Author: Wenyi Tang
#  Email: wenyi.tang@intel.com
#  Update Date: 2019/4/3 下午5:03

from setuptools import find_packages
from setuptools import setup

VERSION = '0.8.2'

REQUIRED_PACKAGES = [
  'numpy',
  'scipy',
  'scikit-image',
  'matplotlib',
  'Pillow',
  'pypng',
  'pytest',
  'PyYAML',
  'psutil',
  'tqdm',
  'h5py',
  'easydict >= 1.9',
  'google-api-python-client',
  'oauth2client',
]

try:
  import torch

  # tensorboardX is absorbed into 1.1.0 now
  REQUIRED_PACKAGES.extend([
    'torch >= 1.1.0',
    'torchvision',
  ])
except ImportError:
  pass

setup(
  name='VSR',
  version=VERSION,
  description='Video Super-Resolution Framework',
  url='https://github.com/LoSealL/VideoSuperResolution',
  packages=find_packages(),
  install_requires=REQUIRED_PACKAGES,
  license='MIT',
  author='Wenyi Tang',
  author_email='wenyitang@outlook.com',
  keywords="super-resolution sr vsr tensorflow pytorch",
)