diff --git a/doc/doc_ch/whl.md b/doc/doc_ch/whl.md
index 0c969988a649ba6cea0115abd1ba0aa5b19ea892..edacb5b2d804d85fd593f4a6c439fc223a8f5e4b 100644
--- a/doc/doc_ch/whl.md
+++ b/doc/doc_ch/whl.md
@@ -297,6 +297,7 @@ paddleocr --image_dir http://n.sinaimg.cn/ent/transform/w630h933/20171222/o111-f
 ### 4.2 numpy数组
 仅通过代码使用时支持numpy数组作为输入
 ```python
+import cv2
 from paddleocr import PaddleOCR, draw_ocr
 # Paddleocr目前支持中英文、英文、法语、德语、韩语、日语，可以通过修改lang参数进行切换
 # 参数依次为`ch`, `en`, `french`, `german`, `korean`, `japan`。
diff --git a/doc/doc_en/whl_en.md b/doc/doc_en/whl_en.md
index 2d68845b18ef8937c626179c2e0a746201341bec..79bc9b0bc5c5f7f33682d5829e5b613b278a4964 100644
--- a/doc/doc_en/whl_en.md
+++ b/doc/doc_en/whl_en.md
@@ -305,6 +305,7 @@ paddleocr --image_dir http://n.sinaimg.cn/ent/transform/w630h933/20171222/o111-f
 Support numpy array as input only when used by code
 
 ```python
+import cv2
 from paddleocr import PaddleOCR, draw_ocr
 ocr = PaddleOCR(use_angle_cls=True, lang="ch") # need to run only once to download and load model into memory
 img_path = 'PaddleOCR/doc/imgs/11.jpg'