From b0aecea2f2d5c60af4936eea0566736d3f342280 Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Mon, 16 Dec 2019 14:31:11 +0800
Subject: [PATCH] Add opencv dependency

---
 requirements.txt | 1 +
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 5011ce56..1ae30f64 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 45df3f30..5e96e96b 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:
-- 
GitLab