huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the imagenet specific bits to a separate file. (#571)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-23 16:42:09 +01:00
committed by GitHub
parent 7478dda255
commit eedd85ffa7
3 changed files with 1026 additions and 1024 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/dinov2/main.rs
View File

@ -306,7 +306,7 @@ pub fn main() -> anyhow::Result<()> {
let device = candle_examples::device(args.cpu)?; let device = candle_examples::device(args.cpu)?;
let image = candle_examples::load_image224(args.image)?; let image = candle_examples::imagenet::load_image224(args.image)?;
println!("loaded image {image:?}"); println!("loaded image {image:?}");
let model_file = match args.model { let model_file = match args.model {
@ -331,7 +331,7 @@ pub fn main() -> anyhow::Result<()> {
for &(category_idx, pr) in prs.iter().take(5) { for &(category_idx, pr) in prs.iter().take(5) {
println!( println!(
"{:24}: {:.2}%", "{:24}: {:.2}%",
candle_examples::IMAGENET_CLASSES[category_idx], candle_examples::imagenet::CLASSES[category_idx],
100. * pr 100. * pr
); );
} }

1023
candle-examples/src/imagenet.rs Normal file
View File

File diff suppressed because it is too large Load Diff

1023
candle-examples/src/lib.rs
View File

File diff suppressed because it is too large Load Diff