diff --git a/requirements.txt b/requirements.txt
index 0840aad..ef0b752 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,3 +7,4 @@ PyQt5==5.15.7
 PyYAML==6.0
 torch==1.8.1
 torchvision==0.9.1
+pycocotools