Update README.md

kyegomez · Mar 30, 2024 · 321a07d · 321a07d
1 parent 122a8b3
commit 321a07d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -37,10 +37,18 @@ print(out.shape)
 ```
 
 
+## Train
+Dependencies: download and extract the datasets through wget
+`wget http://images.cocodataset.org/zips/train2017.zip -O coco_train2017.zip`
+`wget http://images.cocodataset.org/zips/val2017.zip -O coco_val2017.zip`
+`wget http://images.cocodataset.org/annotations/annotations_trainval2017.zip -O coco_ann2017.zip`
+
+`python3 train.py`
+
 # License
 MIT
 
 # Todo
 - [ ] Add paper link
 - [ ] Add citation bibtex
-- [ ] cleanup
+- [ ] cleanup