From 10eaa6e8c70aa0309240f99ade15cab5ebd0691a Mon Sep 17 00:00:00 2001 From: kartikdutt18 Date: Fri, 25 Sep 2020 18:46:47 +0530 Subject: [PATCH 1/2] YOLO Docs --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 447fa5d5..823213e7 100644 --- a/README.md +++ b/README.md @@ -199,6 +199,7 @@ Currently model-zoo project has the following models implemented: | --- | --- | --- | --- | | Darknet 19 | DarkNet, HeInitialization, 19> darknet19({imageDepth, imageWidth, imageHeight}, numClasses)| ImageNet |[YOLO9000](https://pjreddie.com/media/files/papers/YOLO9000.pdf)| | Darknet 53 | DarkNet, HeInitialization, 53> darknet19({imageDepth, imageWidth, imageHeight}, numClasses)| ImageNet |[YOLOv3](https://pjreddie.com/media/files/papers/YOLOv3.pdf)| +| YOLOv1 | YOLO<> yolov1({imageDepth, imageWidth, imageHeight}, numClasses)| VOC |[YOLO](https://arxiv.org/pdf/1506.02640.pdf)| All models can be included as shown below : ```cpp From 27cbbf60aac1c820fd392e24cde0d2f25ce7478b Mon Sep 17 00:00:00 2001 From: kartikdutt18 Date: Thu, 1 Oct 2020 00:48:55 +0530 Subject: [PATCH 2/2] Add docs --- README.md | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 823213e7..3438bf03 100644 --- a/README.md +++ b/README.md @@ -54,6 +54,7 @@ main mlpack repository.)_ spirit) >= 1.58 CMake >= 3.3.2 ensmallen >= 2.10.0 + opencv >= 2.x To install mlpack refer to the [installation guide](https://www.mlpack.org/docs.html) that's available in the mlpack @@ -173,7 +174,7 @@ For all datasets that we support we provide, We preprocess them internally. We a provide access to preprocessor functions for standard datasets incase one needs to apply them to their datasets. -They can simply be called as follows by calling static functions of ProProcess class i.e. +They can simply be called as follows by calling static functions of PreProcess class i.e. PreProcess::SupportedDatasetName ```cpp @@ -206,4 +207,12 @@ All models can be included as shown below : #include ``` +### 7. Visualization Tools + +We use open cv to visulize plots such as bounding boxes. The visualization functions are static functions that are a part of `VisualizationTools` class. + +`VisualizationTools::FunctionName(paramters);` + +For more details, take a look at our wiki page. + For more information about usage, take a look at our wiki page.