huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +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 image = candle_examples::load_image224(args.image)?;
let image = candle_examples::imagenet::load_image224(args.image)?;
println!("loaded image {image:?}");
let model_file = match args.model {
@ -331,7 +331,7 @@ pub fn main() -> anyhow::Result<()> {
for &(category_idx, pr) in prs.iter().take(5) {
println!(
"{:24}: {:.2}%",
candle_examples::IMAGENET_CLASSES[category_idx],
candle_examples::imagenet::CLASSES[category_idx],
100. * pr
);
}