Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add docs for bounding boxes and YOLO. #37

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Add docs for bounding boxes and YOLO. #37

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
10eaa6e
YOLO Docs
kartikdutt18 Sep 25, 2020
27cbbf6
Add docs
kartikdutt18 Sep 30, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 11 additions & 1 deletion README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand All @@ -199,10 +200,19 @@ Currently model-zoo project has the following models implemented:
| --- | --- | --- | --- |
| Darknet&nbsp;19 | DarkNet<NegativeLogLikelihood<>, HeInitialization, 19>&nbsp;darknet19({imageDepth, imageWidth, imageHeight}, numClasses)| ImageNet |[YOLO9000](https://pjreddie.com/media/files/papers/YOLO9000.pdf)|
| Darknet&nbsp;53 | DarkNet<NegativeLogLikelihood<>, HeInitialization, 53>&nbsp;darknet19({imageDepth, imageWidth, imageHeight}, numClasses)| ImageNet |[YOLOv3](https://pjreddie.com/media/files/papers/YOLOv3.pdf)|
| YOLOv1 | YOLO<>&nbsp;yolov1({imageDepth, imageWidth, imageHeight}, numClasses)| VOC |[YOLO](https://arxiv.org/pdf/1506.02640.pdf)|

All models can be included as shown below :
```cpp
#include <models/Model-ClassName/Model_ClassName.hpp>
```

### 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.