diff --git a/requirements.txt b/requirements.txt index 5011ce56b62d63cbf8578bed5473deb8608917cb..1ae30f64ccfc3e1994e2d4f63edec51951487b69 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,6 +14,7 @@ cma == 2.7.0 sentencepiece nltk colorlog +opencv-python # numpy no longer support python2 in version 1.17 and above numpy ; python_version >= "3" diff --git a/setup.py b/setup.py index 45df3f309ee8d266cfc3aa8d0968000b53312b00..5e96e96b8eca3c980b5308c34747abb36ddef5d7 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,7 @@ max_version, mid_version, min_version = python_version() REQUIRED_PACKAGES = [ 'six >= 1.10.0', 'protobuf >= 3.6.0', 'pyyaml', 'Pillow', 'requests', 'tb-paddle', 'tensorboard >= 1.15', 'cma == 2.7.0', 'flask >= 1.1.0', - 'sentencepiece', 'nltk', 'colorlog' + 'sentencepiece', 'nltk', 'colorlog', 'opencv-python' ] if max_version < 3: