Commit Graph

13 Commits

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