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

[segment-anything] Print IOU values to help with debugging (#999)

Browse Source
This commit is contained in:
GeauxEric
2023-09-29 21:44:42 -07:00
committed by GitHub
parent d188d6a764
commit 34842fb234
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/segment-anything/main.rs
View File

@ -126,7 +126,7 @@ pub fn main() -> anyhow::Result<()> {
start_time.elapsed().as_secs_f32() start_time.elapsed().as_secs_f32()
); );
println!("mask:\n{mask}"); println!("mask:\n{mask}");
println!("iou_predictions: {iou_predictions:?}"); println!("iou_predictions: {iou_predictions}");
let mask = (mask.ge(args.threshold)? * 255.)?; let mask = (mask.ge(args.threshold)? * 255.)?;
let (_one, h, w) = mask.dims3()?; let (_one, h, w) = mask.dims3()?;