From 5f41aa92c4fd8b64a714d076db695daa5e656056 Mon Sep 17 00:00:00 2001
From: Yuren Cong <47991543+yrcong@users.noreply.github.com>
Date: Thu, 3 Feb 2022 14:28:14 +0100
Subject: [PATCH] Update README.md

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 66845b0..58c40f2 100644
--- a/README.md
+++ b/README.md
@@ -41,9 +41,10 @@ conda install matplotlib
 ## Training/Evaluation on Visual Genome
 If you want to **train/evaluate** RelTR on Visual Genome, you need a little more preparation:
 
-a) Scipy (we used 1.5.2) is required. 
+a) Scipy (we used 1.5.2) and pycocotools are required. 
 ```
 conda install scipy
+pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
 ```
 
 b) Download the annotations of [Visual Genome (in COCO-format)](https://drive.google.com/file/d/1aGwEu392DiECGdvwaYr-LgqGLmWhn8yD/view?usp=sharing) and unzip it in the ```data/``` forder.