Commit Graph

28 Commits

Author SHA1 Message Date
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 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 826da4037e
Update Cargo.toml 2024-09-03 19:03:25 +08:00
Jamjamjon 9b969b9203
Fix polygon drawing
* Support auto download ort feature
* Fix polygon drawing
2024-09-03 00:01:34 +08:00
Jamjamjon f6755a8be4
Bump the version to 0.0.12
* Add sapiens-seg model
2024-08-31 17:10:36 +08:00
Jamjamjon f25f5cf2b5
Bump the version to v0.0.11
- ONNXRuntime -> 1.19.x
- CUDA -> 12.x
- TensorRT -> 10.x
2024-08-27 20:09:15 +08:00
Jamjamjon b81b5e3cf5
Add GroundingDINO (#30) 2024-08-09 19:06:30 +08:00
Jamjamjon 53d14ee2fb
Add `Xs`, a wrapper over `Vec<X>` (#29) 2024-08-03 18:03:35 +08:00
Jamjamjon c4c240ef75
Update Cargo.toml 2024-08-01 21:47:34 +08:00
Jamjamjon 1d74085158
0.0.7 (#27) 2024-07-31 21:27:41 +08:00
Jamjamjon 0901ab3e3c
Update Cargo.toml 2024-07-24 00:39:48 +08:00
Jamjamjon 9a730ccc3e
Fix rgba bug when resizing (#26) 2024-07-24 00:30:27 +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 25d9088e2e
Update Cargo.toml 2024-07-01 17:14:06 +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
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 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
Jamjamjon c05836f02e
Clean up 2024-04-14 18:11:09 +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 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