candle

mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00

Files

shua 6056fd5c90 onnx: fix pad, unsqueeze (#2317 )

* onnx: fix pad, unsqueeze

both implementations have off-by-one errors:
- Pad 'reflect' cycle for eg `dim==3` is `[0,1,2,1]` which has length of
  4 (or `dim*2 - 2`) not 5 (current code `dim*2 - 1`)
- Unsqueeze(-1) for tensor with `dim==3` should be 3 (ie `dim+index+1`)
  not 2 (ie currently `dim+index`)

in addition, Pad is incorrectly calculating the starting padding.
If we want to pad out 2 elements to the start, and we have this cycle
of indices of length 6, then we should skip 4 elements, but currently
we skip 2. A more visual representation of what's going on is below:

```
pad_start: 2
data:      [a,b,c,d]
indices:   [0, 1, 2, 3, 2, 1, 0, 1, 2, 3, 2, 1, 0, ..] // zigzag between 0..4
actual:    skip [ c  d| c  b  a  b]
expected:  ~  skip  ~ [ c  b| a  b  c  d]
```

The values between `[` and `|` are padding and the values between
`|` and `]` in the example should match the original data being padded.

* Fix clippy lints.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>

2024-07-23 23:10:57 +02:00

src

onnx: fix pad, unsqueeze (#2317 )

2024-07-23 23:10:57 +02:00

build-lib.sh

Also fix the aspect ratio in the wasm example. (#556 )

2023-08-22 22:20:08 +01:00

Cargo.toml

Update gloo requirement from 0.8 to 0.11 (#1558 )

2024-01-10 16:27:20 +01:00

index.html

[WIP] Improve Yolo WASM UI example (#591 )

2023-08-26 11:40:41 +01:00

lib-example.html

minor UI fixes (#856 )

2023-09-15 06:30:50 +01:00

README.md

Add Yolo Pose to JS Example (#684 )

2023-08-31 06:32:57 +01:00

yoloWorker.js

force model cache (#751 )

2023-09-06 05:53:31 +02:00

README.md

Running Yolo Examples

Here, we provide two examples of how to run YOLOv8 using a Candle-compiled WASM binary and runtimes.

Pure Rust UI

To build and test the UI made in Rust you will need Trunk From the candle-wasm-examples/yolo directory run:

Download assets:

wget -c https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/candle/examples/bike.jpeg
wget -c https://huggingface.co/lmz/candle-yolo-v8/resolve/main/yolov8s.safetensors

Run hot reload server:

trunk serve --release --public-url / --port 8080

Vanilla JS and WebWorkers

To build and test the UI made in Vanilla JS and WebWorkers, first we need to build the WASM library:

sh build-lib.sh

This will bundle the library under ./build and we can import it inside our WebWorker like a normal JS module:

import init, { Model, ModelPose } from "./build/m.js";

The full example can be found under ./lib-example.html. All needed assets are fetched from the web, so no need to download anything. Finally, you can preview the example by running a local HTTP server. For example:

python -m http.server

Then open http://localhost:8000/lib-example.html in your browser.