mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-12-03 02:58:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add onnx build script

Browse Source
This commit is contained in:
jamjamjon
2024-05-13 20:24:57 +08:00
parent fc970fc117
commit b9c077814b
13 changed files with 893 additions and 1086 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/clip/main.rs
View File

@@ -39,7 +39,10 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
let feats_image = model.encode_images(&images).unwrap();
// use image to query texts
let matrix = feats_image.dot2(&feats_text)?;
let matrix = match feats_image.embedding() {
Some(x) => x.dot2(feats_text.embedding().unwrap())?,
None => continue,
};
// summary
for i in 0..paths.len() {

BIN
examples/dinov2/images/bus.jpg
View File

Binary file not shown.
Side by Side

Before

Width:  |  Height:  |  Size: 134 KiB

8
examples/dinov2/main.rs
View File

@@ -1,4 +1,4 @@
use usls::{models::Dinov2, Options};
use usls::{models::Dinov2, DataLoader, Options};
fn main() -> Result<(), Box<dyn std::error::Error>> {
// build model
@@ -7,8 +7,10 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
.with_i00((1, 1, 1).into())
.with_i02((224, 224, 224).into())
.with_i03((224, 224, 224).into());
let _model = Dinov2::new(options)?;
println!("TODO...");
let mut model = Dinov2::new(options)?;
let x = vec![DataLoader::try_read("./examples/dinov2/images/1.jpg")?];
let y = model.run(&x)?;
println!("{y:?}");
// query from vector
// let ys = model.query_from_vec(