Commit Graph

40 Commits

Author SHA1 Message Date
Jamjamjon
e2347353ba Add SmolVLM model (#65)
* Add SmolVLM model
2025-02-08 00:28:35 +08:00
jamjamjon
bdd77a6d21 Add svtr_en model 2025-01-25 01:53:13 +08:00
Jamjamjon
94fa832359 Add Owlv2 model (#59) 2025-01-22 23:49:51 +08:00
Jamjamjon
b3072e8da3 Bug Fixes: Increased the number of available releases and fixed the Trocr bug. (#58) 2025-01-22 14:55:12 +08:00
Jamjamjon
475a680703 Add moondream2
* Add moondream2

* Update README.md
2025-01-20 21:37:54 +08:00
jamjamjon
f511a475ac Using rayon to accelerate DB post-processing 2025-01-13 22:42:43 +08:00
Jamjamjon
0f2d84b8c5 🐍 v0.1.0 (#53) 2025-01-12 16:59:57 +08:00
Jamjamjon
447889028e Add Apple ml-depth-pro model 2024-10-10 00:26:26 +08:00
Jamjamjon
1d596383de Add support for restricting detection classes (#45)
* Add support for restricting detection classes in `Options`
2024-10-05 17:49:08 +08:00
Jamjamjon
0102c15687 Minor fixes 2024-10-01 09:37:46 +08:00
Jamjamjon
0609dd1f1d Add YOLOv11
* Add YOLOv11
2024-09-30 22:43:34 +08:00
Jamjamjon
6ace97f09f v0.0.16: Add Viewer for image show
* Add `Viewer` for image-show

* Add .`with_ixx()` to simplify model building with `Options`

* Update `MinOptMax`
2024-09-28 09:51:33 +08:00
Jamjamjon
f0fd4936e8 Add florence2 model
* Add florence2-base model for all tasks

* Update annotator.rs
2024-09-21 20:23:42 +08:00
Jamjamjon
0adddd3bbd 0.0.14: DataLoader now support video and streaming
- Added `Hub` for resource management
- Updated `DataLoader` to support video and streaming
- Updated `CI`
- Replaced `println!` with `tracing` for logging
2024-09-16 10:41:16 +08:00
Jamjamjon
f6755a8be4 Bump the version to 0.0.12
* Add sapiens-seg model
2024-08-31 17:10:36 +08:00
Jamjamjon
b81b5e3cf5 Add GroundingDINO (#30) 2024-08-09 19:06:30 +08:00
Jamjamjon
46a4456a38 Add SAM2 and ONNX (#28) 2024-08-01 17:26:06 +08:00
Jamjamjon
1d74085158 0.0.7 (#27) 2024-07-31 21:27:41 +08:00
jamjamjon
56aed9e50c Update ort 2024-07-14 10:59:44 +08:00
Jamjamjon
edc3a8897c 0.0.5 (#24)
* Using Rayon to accelarate YOLO post-processing

* Refactor YOLO with outputs format

* Optimize `conf * clss` for yolov5 v6 v7

* Add depth-anything-v2

* Update README.md

* Update CHANGELOG.md
2024-07-12 19:46:48 +08:00
Jamjamjon
a5141a53be Accelerate model pre-processing and post-processing (#23)
* Add X struct to handle input and preprocessing

*Add Ops struct to manage common operations

* Use SIMD (fast_image_resize) to accelerate model pre-processing and post-processing
2024-06-30 15:19:34 +08:00
oatiz
5f6b814090 Add an option to adjust the line width for Bbox edges (#22)
* feat: add bbox thickness option

* feat: add thickness threshold and side option

* fix: clippy

* Minor adjustments

---------

Co-authored-by: jamjamjon <xxyydzml@outlook.com>
2024-06-26 00:00:36 +08:00
Jamjamjon
311d49f5b7 Minor adjustments to YOLO (#17) 2024-06-02 12:49:42 +08:00
Jamjamjon
20577e4242 Adjust outputs (#16) 2024-06-01 16:58:17 +08:00
oatiz
a65e2aff7b feat: support yolov10 (#15)
Signed-off-by: oatiz <iamoatiz@gmail.com>
2024-06-01 16:42:38 +08:00
jamjamjon
c44d80df78 Revert "Add onnx build script"
This reverts commit b9c077814b.
2024-05-13 20:59:05 +08:00
jamjamjon
b9c077814b Add onnx build script 2024-05-13 20:24:57 +08:00
Jamjamjon
fc970fc117 Update ort and improve the speed of preprocessing
* Add onnx proto

* Update ort to 2.0.0-rc.2

* Improve the speed of resizing

* Fix yolo-seg bug

* Update README.md
2024-05-12 18:12:34 +08:00
Jamjamjon
371a08011f Add MODNet model (#11)
* Add MODNet for portrait matting

* Minor fixes

* Move assets to home directory

* Add colormap

* ci

* Update README.md
2024-04-30 15:26:53 +08:00
Jamjamjon
e614ca4136 Add Depth-Anything model (#10)
* Add Depth-Anything model
2024-04-25 00:41:29 +08:00
Jamjamjon
beda8ef803 Add YOLOv8-OBB and some bug fixes (#9)
* Add YOLOv8-Obb & Refactor outputs

* Update README.md
2024-04-21 17:06:58 +08:00
oatiz
91049fc18a fix: svtr decode bug (#8) 2024-04-17 20:11:11 +08:00
oatiz
a86000d107 fix: svtr deduplication bug (#7) 2024-04-16 18:59:42 +08:00
Jamjamjon
c05836f02e Clean up 2024-04-14 18:11:09 +08:00
Jamjamjon
51b75e9a21 Add YOLOPv2 & Face-Parsing model (#3)
* Add YOLOP and face parsing model
2024-04-14 15:15:59 +08:00
Jamjamjon
ead175234c Add RTMO model for keypoint detection (#2)
* Add RTMO
2024-04-08 20:45:38 +08:00
Jamjamjon
a0d410b46d Dev (#1)
* Update imageproc crates

* Add top-p method for sampling

* Add SVTR for text recognition & bug fix
2024-04-06 16:16:53 +08:00
jamjamjon
ce9a416b71 Add DB model for text detection 2024-03-31 02:14:53 +08:00
jamjamjon
a5cee66dfd Add query method for dinov2 and adjust DataLoader 2024-03-30 15:54:47 +08:00
jamjamjon
af934086bb Initial 2024-03-29 15:54:24 +08:00