Commit Graph

29 Commits

Author SHA1 Message Date
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
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