Commit Graph

61 Commits

Author SHA1 Message Date
jamjamjon 4e932c4910 Bump the version to 0.0.20 2024-12-03 19:37:34 +08:00
Collide 2785b090c6
upgrade ort to v2.0.0-rc.9 (#52) 2024-12-03 19:16:23 +08:00
Jamjamjon 57db14ce5d
Update README.md 2024-10-10 00:30:52 +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 64dc804a13
Update README.md 2024-09-30 22:48:07 +08:00
Jamjamjon 0609dd1f1d
Add YOLOv11
* Add YOLOv11
2024-09-30 22:43:34 +08:00
Jamjamjon 2cb9e57fc4
Update README.md 2024-09-28 10:49:06 +08:00
Jamjamjon f2c4593672
Update README.md 2024-09-28 10:10:05 +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 5057c20565
Update README.md 2024-09-16 11:05:53 +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 46a4456a38
Add SAM2 and ONNX (#28) 2024-08-01 17:26:06 +08:00
Jamjamjon 451aa8cc7b
Update README.md 2024-07-31 23:16:35 +08:00
Jamjamjon d9f2d6382b
Update README.md 2024-07-31 21:28:52 +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 7740761345
Fix CPU ep 2024-07-17 14:25:28 +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 f6fec32c7b Docs update 2024-07-01 16:53:36 +08:00
Jamjamjon 20d2edce6a
Update README.md 2024-06-30 23:40:20 +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 6c72374ff6
Create preprocess.cu 2024-06-14 22:06:53 +08:00
Jamjamjon 510d10f18a
Update rtmo.rs 2024-06-06 14:46:12 +08:00
Jamjamjon 1db1dfa4bd
Update README.md 2024-06-06 13:18:23 +08:00
young.day 4fa1ed5824
issued:attempt to subtract with overflow (#20) 2024-06-06 11:01:25 +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 246e6091cd
Add decimal_places for annotator (#13) 2024-05-04 12:48:55 +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 9697f4d5b0
Update README.md 2024-04-25 11:18:52 +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