Commit message (Collapse) | Author | Age | Files | Lines | |
---|---|---|---|---|---|
* | onnx: fix pad, unsqueeze (#2317) | shua | 2024-07-23 | 1 | -1/+1 |
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | * 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> | ||||
* | Remove some unusued bits. (#1067) | Laurent Mazare | 2023-10-09 | 1 | -1/+0 |
| | |||||
* | [segment-anything] add multi point logic for demo site (#1002) | lichin-lin | 2023-10-01 | 1 | -12/+24 |
| | | | | | * [segment-anything] add multi point logic for demo site * [segment-anything] remove libs and update functions | ||||
* | Add negative prompts to segment-anything. (#1000) | Laurent Mazare | 2023-09-30 | 1 | -1/+1 |
| | |||||
* | [segment-anything] Support multi-point as the prompt input (#945) | GeauxEric | 2023-09-25 | 1 | -1/+1 |
| | | | | | | | | | | | | | | * [sam] Support multi-point prompts * [segment-anything] Pass points by reference * [segment-anything] Update example code and image * Fix clippy lint. --------- Co-authored-by: Yun Ding <yunding@nvidia.com> Co-authored-by: laurent <laurent.mazare@gmail.com> | ||||
* | Depreate the VarBuilder::from_safetensors function. (#951) | Laurent Mazare | 2023-09-24 | 1 | -3/+2 |
| | |||||
* | Update README + SAM (#866) | Radamés Ajna | 2023-09-16 | 1 | -3/+2 |
| | | | | | * use serde-wasm-bindgen, faster serialization * update readme with demos | ||||
* | Add SAM UI Demo (#854) | Radamés Ajna | 2023-09-15 | 1 | -2/+20 |
| | | | | | | | | | | | | | * fix tensor flattening * send image data back * sam ui worker example * SAM example * resize container * no need for this | ||||
* | Return the low res mask in the wasm segment-anything module. (#798) | Laurent Mazare | 2023-09-10 | 1 | -1/+11 |
| | | | | | * Return the low res mask. * Add some validations. | ||||
* | Add a wasm module for the segment anything example. (#797) | Laurent Mazare | 2023-09-10 | 2 | -0/+132 |