Skip to content

Commit

Permalink
add yolov10
Browse files Browse the repository at this point in the history
  • Loading branch information
OneMagicKey committed May 28, 2024
1 parent 635fb36 commit 0d04f28
Show file tree
Hide file tree
Showing 6 changed files with 103,596 additions and 11 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,22 +8,22 @@

<h2 align="center"><img width="768" src="example/example.png"></h2>

A telegram bot for object detection and instance segmentation using YOLOv5/YOLOv8,
A telegram bot for object detection and instance segmentation using YOLOv5/YOLOv8/YOLOv10,
implemented in Python + OpenCV + ONNXRuntime.

Additionally, the bot can be deployed on the [Render](https://render.com) cloud platform
for free.

The webhook version of the running bot requires around 380 MB of RAM and includes 2
**quantized** models: `yolov8s` and `yolov8s-seg`. The pooling version includes 4
**quantized** models: `yolov5s`, `yolov8s`, `yolov5s-seg`, and `yolov8s-seg`,
and requires around 560 MB of RAM.
**quantized** models: `yolov10s` and `yolov8s-seg`. The pooling version includes 5
**quantized** models: `yolov5s`, `yolov8s`, `yolov10s`,`yolov5s-seg`, and `yolov8s-seg`,
and requires around 600 MB of RAM.

## Features

- [x] ENG / RU languages
- [x] Object Detection / Instance Segmentation
- [x] Quantized YOLO models
- [x] Quantized YOLOv5, YOLOv8 and YOLOv10 models
- [x] Support webhooks to deploy as a webservice on the Render platform ([webhooks-render](https://github.com/OneMagicKey/Yolov8-ONNX-telegram-bot/blob/webhooks-render/) branch)
- [x] Support pooling to run bot locally ([master](https://github.com/OneMagicKey/Yolov8-ONNX-telegram-bot/blob/master/) branch)

Expand Down
1 change: 1 addition & 0 deletions src/bot.py
Original file line number Diff line number Diff line change
Expand Up @@ -366,6 +366,7 @@ async def main():


model_list = [
ModelInfo("detection", "yolov10s", (640, 640), 10),
ModelInfo("detection", "yolov8s", (640, 640), 8),
ModelInfo("detection", "yolov5s", (640, 640), 5),
ModelInfo("segmentation", "yolov8s-seg", (640, 640), 8),
Expand Down
Loading

0 comments on commit 0d04f28

Please sign in to comment.