diff --git a/requirements.txt b/requirements.txt
index eee51aa..a291d7e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,3 +10,4 @@ shapely
 torch
 torchvision
 tqdm
+matplotlib