From 3ea83056d7b4791f4ff8bf2a62a00f4d447b9c19 Mon Sep 17 00:00:00 2001 From: ControlNet Date: Tue, 23 Apr 2024 20:36:22 +1000 Subject: [PATCH 1/2] fix AR calculation when the num proposals is less than the max_proposals --- src/tensorneko_lib/evaluation/loc_1d.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tensorneko_lib/evaluation/loc_1d.rs b/src/tensorneko_lib/evaluation/loc_1d.rs index 18461c6..8d5c926 100644 --- a/src/tensorneko_lib/evaluation/loc_1d.rs +++ b/src/tensorneko_lib/evaluation/loc_1d.rs @@ -215,7 +215,7 @@ fn calc_ar_values( labels: &Array2, fps: f32, ) -> ArrayBase, Ix3> { - let max_proposals = *n_proposals.iter().max().unwrap(); + let max_proposals = *n_proposals.iter().max().unwrap().min(proposals.nrows()); let mut proposals = proposals.slice(s![..max_proposals, ..]).to_owned(); if proposals.is_empty() { From f256cbc94db0428d7f8a629d0618a03c7cae2e80 Mon Sep 17 00:00:00 2001 From: ControlNet Date: Tue, 23 Apr 2024 20:50:41 +1000 Subject: [PATCH 2/2] fix AR calculation problem --- src/tensorneko_lib/evaluation/loc_1d.rs | 3 ++- version.txt | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/tensorneko_lib/evaluation/loc_1d.rs b/src/tensorneko_lib/evaluation/loc_1d.rs index 8d5c926..4b3fa6f 100644 --- a/src/tensorneko_lib/evaluation/loc_1d.rs +++ b/src/tensorneko_lib/evaluation/loc_1d.rs @@ -215,7 +215,8 @@ fn calc_ar_values( labels: &Array2, fps: f32, ) -> ArrayBase, Ix3> { - let max_proposals = *n_proposals.iter().max().unwrap().min(proposals.nrows()); + let max_proposals = *n_proposals.iter().max().unwrap(); + let max_proposals = max_proposals.min(proposals.nrows()); let mut proposals = proposals.slice(s![..max_proposals, ..]).to_owned(); if proposals.is_empty() { diff --git a/version.txt b/version.txt index a70b636..0010ffe 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -0.3.13 \ No newline at end of file +0.3.14 \ No newline at end of file