diff --git a/requirements.txt b/requirements.txt
index e91d0f8..60d7c4a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,7 +14,7 @@ pandas==0.22.0
 Pillow==5.1.0
 pycocotools==2.0.0
 pydot_ng==1.0.0
-PyYAML==3.12
+PyYAML==5.1
 scikit-learn==0.19.2
 scikit_image==0.13.1
 scipy==1.0.0