diff --git a/requirements.txt b/requirements.txt
index 08c696bb..7b631ebe 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,6 +3,7 @@
 numpy == 1.17
 opencv-python >= 4.1
 torch >= 1.5
+torchvision
 matplotlib
 pycocotools
 tqdm