Skip to content

Commit

Permalink
Merge pull request #33 from saaresearch/master
Browse files Browse the repository at this point in the history
update
  • Loading branch information
aimspot authored Jun 25, 2024
2 parents 3cf90f3 + d217aa5 commit 3dd0c0e
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 41 deletions.
50 changes: 9 additions & 41 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,69 +3,37 @@
[![PythonVersion](https://img.shields.io/badge/python-3.8%20%7C%203.9%20%7C%203.10-blue)](https://pypi.org/project/scikit-learn/)

[![Documentation Status](https://readthedocs.org/projects/odrs-test/badge/?version=latest)](https://odrs-test.readthedocs.io/en/latest/?badge=latest)

[![wiki](https://img.shields.io/badge/wiki-latest-blue)](http://www.wiki.odrs.space)
<div align="center">
<p>
<a align="center" href="https://github.com/saaresearch/ODRS" target="_blank">
<img width="70%" src="docs/img/logo.png">
<img width="70%" src="docs/img/logo_new.png">
</a>
</p>
<div style='display: block;'>
<a href="https://itmo.ru/">
<img width="10%" src="docs/img/itmo.png" alt="Acknowledgement to ITMO">
</a>
<a href="">
<a href="https://colab.research.google.com/drive/1iTx37IwvGyms82626ALYqZfdEbhGmJll?usp=sharing">
<img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab">
</a>
</div>
</div>
ODRS - it an open source recommendation system for training object detection models. Our system allows you to choose the most
<br></br>

**ODRS** - it an open source recommendation system for training object detection models. Our system allows you to choose the most
profitable existing object recognition models based on user preferences and data. In addition to choosing the
architecture of the model, the system will help you start training and configure the environment.

The proposed recommendation system consists of several components that interact to generate recommendations for machine learning pipelines.
<div align="center">
<img src="docs/img/system.jpg" width="400">
<img src="docs/img/prew.gif" width="640" height="360">
</div>
External parameters (received from users and third-party resources):
* Dataset: Represents input data (video frames) and associated metadata (e.g. image size, quality, number of objects).
* Model: Framework provides an opportunity to train the most popular object recognition models (including setting up the environment
and choosing the architecture of a specific model). Considered two-stage detectors models such as Faster R-CNN and Mask R-CNN as
well as one-stage detectors such as SSD and YOLO (including families v5, v7, v8).
<br/>
<br>
<div align="center">
<img src="docs/img/model.png" width="400">
</div>

Internal components:

* ***RecommendationEngine***: generates recommendations based on user data and dataset characteristics.
<br/>
<br>
<div align="center">
<img src="docs/img/rec_alg.jpg" width="400">
</div>

The recommendation algorithm is based on production rules. The primary set of rules (knowledge base) is formed on
the basis of the results of the analysis of scientific sources and standard data sets, but also empirical processing
of data sets from specific industries.
The main criteria for drawing up the rules were chosen:

1. Dimension of the model
2. The value of metrics (mAP, Recall, Accuracy) for selected datasets
3. The speed of the model on GPU and CPU
4. Supported image format and dimension


* ***Training*** - Training of models proposed by the system
<br/>
<br>
The proposed recommendation system consists of several components that interact to generate recommendations for machine learning pipelines.
<div align="center">
<img src="docs/img/train.jpg" width="400">
<img src="docs/img/alg.gif" width="853" height="480">
</div>

* ***Evaluation*** – evaluation of the quality of training models


## Contents
Expand Down
Binary file added docs/img/alg.gif
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/img/logo_new.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/img/prew.gif
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 3dd0c0e

Please sign in to comment.