huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 04:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* Also fix the aspect ratio in the wasm example.

* Add the yolo lib.

* Update the build script.
This commit is contained in:
Laurent Mazare
2023-08-22 22:20:08 +01:00
committed by GitHub
parent f9ecc84477
commit 7687a0f453
6 changed files with 68 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-wasm-examples/yolo/src/model.rs
View File

@ -5,8 +5,8 @@ use candle_nn::{
};
use image::DynamicImage;
const CONFIDENCE_THRESHOLD: f32 = 0.5;
const NMS_THRESHOLD: f32 = 0.4;
const CONFIDENCE_THRESHOLD: f32 = 0.25;
const NMS_THRESHOLD: f32 = 0.45;
// Model architecture from https://github.com/ultralytics/ultralytics/issues/189
// https://github.com/tinygrad/tinygrad/blob/master/examples/yolov8.py