diff --git a/requirements.txt b/requirements.txt
index 81fea3f..f71f07e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ numpy==1.17.0
 opencv-python==4.1.0.25
 pillow>=6.2.0
 protobuf==3.9.0
-PyYAML==5.1.1
+PyYAML==5.4
 scipy==1.3.0
 six==1.12.0
 tensorboardX==1.8