From 0506a2a87865db4121ecbf7bccd03482440bf549 Mon Sep 17 00:00:00 2001 From: Aladdin Persson Date: Mon, 8 Mar 2021 23:57:45 +0100 Subject: [PATCH] Update README.md --- ML/Pytorch/object_detection/YOLOv3/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ML/Pytorch/object_detection/YOLOv3/README.md b/ML/Pytorch/object_detection/YOLOv3/README.md index 8583c95..e38d783 100644 --- a/ML/Pytorch/object_detection/YOLOv3/README.md +++ b/ML/Pytorch/object_detection/YOLOv3/README.md @@ -14,7 +14,7 @@ $ pip install requirements.txt Available on Kaggle: [link](https://www.kaggle.com/dataset/1cf520aba05e023f2f80099ef497a8f3668516c39e6f673531e3e47407c46694) ### Download Pascal VOC dataset -Download the preprocessed dataset from [link](https://www.kaggle.com/aladdinpersson/pascal-voc-yolo-works-with-albumentations). Just unzip this in the main directory. +Download the preprocessed dataset from [link](https://www.kaggle.com/aladdinpersson/pascal-voc-dataset-used-in-yolov3-video). Just unzip this in the main directory. ### Download MS COCO dataset Download the preprocessed dataset from [coming soon](). Just unzip this in the main directory. @@ -26,7 +26,7 @@ Edit the config.py file to match the setup you want to use. Then run train.py | Model | mAP @ 50 IoU | | ----------------------- |:-----------------:| | YOLOv3 (Pascal VOC) | 78.2 | -| YOLOv3 (MS-COCO) | Not done yet | +| YOLOv3 (MS-COCO) | Will probably train on this at some point | The model was evaluated with confidence 0.2 and IOU threshold 0.45 using NMS. @@ -54,4 +54,4 @@ We present some updates to YOLO! We made a bunch of little design changes to mak journal = {arXiv}, year={2018} } -``` \ No newline at end of file +```