diff --git a/setup.py b/setup.py
index 1384a3a..b5e709e 100644
--- a/setup.py
+++ b/setup.py
@@ -19,6 +19,8 @@
     install_requires=[
         'tensorflow-gpu',
         'opencv-python',
+        'matplotlib',
+        'pillow',
         'numpy==1.19.2',
         'scipy==1.6.2',
         'scikit-learn==0.24.2'