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 decimal_places for annotator (#13)

Browse Source
This commit is contained in:
Jamjamjon
2024-05-04 12:48:55 +08:00
committed by GitHub
parent 371a08011f
commit 246e6091cd
5 changed files with 39 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/core/annotator.rs
View File

@@ -14,6 +14,7 @@ pub struct Annotator {
_scale: f32, // Cope with ab_glyph & imageproc=0.24.0
scale_dy: f32,
saveout: Option<String>,
decimal_places: usize,
// About mbrs
without_mbrs: bool,
@@ -65,6 +66,7 @@ impl Default for Annotator {
scale_dy: 28.,
polygons_alpha: 179,
saveout: None,
decimal_places: 4,
without_bboxes: false,
without_bboxes_conf: false,
without_bboxes_name: false,
@@ -98,6 +100,11 @@ impl Default for Annotator {
}
impl Annotator {
pub fn with_decimal_places(mut self, x: usize) -> Self {
self.decimal_places = x;
self
}
/// Plotting BBOXes or not
pub fn without_bboxes(mut self, x: bool) -> Self {
self.without_bboxes = x;
@@ -364,7 +371,11 @@ impl Annotator {
// label
if !self.without_bboxes_name || !self.without_bboxes_conf {
let label = bbox.label(!self.without_bboxes_name, !self.without_bboxes_conf);
let label = bbox.label(
!self.without_bboxes_name,
!self.without_bboxes_conf,
self.decimal_places,
);
self.put_text(
img,
&label,
@@ -395,7 +406,11 @@ impl Annotator {
// label
if !self.without_mbrs_name || !self.without_mbrs_conf {
let label = mbr.label(!self.without_mbrs_name, !self.without_mbrs_conf);
let label = mbr.label(
!self.without_mbrs_name,
!self.without_mbrs_conf,
self.decimal_places,
);
self.put_text(
img,
&label,
@@ -451,7 +466,11 @@ impl Annotator {
if self.with_polygons_name || self.with_polygons_conf {
for polygon in polygons.iter() {
if let Some((x, y)) = polygon.centroid() {
let label = polygon.label(self.with_polygons_name, self.with_polygons_conf);
let label = polygon.label(
self.with_polygons_name,
self.with_polygons_conf,
self.decimal_places,
);
self.put_text(
img,
&label,
@@ -488,7 +507,11 @@ impl Annotator {
// label
if self.with_keypoints_name || self.with_keypoints_conf {
let label = kpt.label(self.with_keypoints_name, self.with_keypoints_conf);
let label = kpt.label(
self.with_keypoints_name,
self.with_keypoints_conf,
self.decimal_places,
);
self.put_text(
img,
&label,

6
src/ys/bbox.rs
View File

@@ -161,7 +161,7 @@ impl Bbox {
self.confidence
}
pub fn label(&self, with_name: bool, with_conf: bool) -> String {
pub fn label(&self, with_name: bool, with_conf: bool, decimal_places: usize) -> String {
let mut label = String::new();
if with_name {
label.push_str(
@@ -174,9 +174,9 @@ impl Bbox {
}
if with_conf {
if with_name {
label.push_str(&format!(": {:.4}", self.confidence));
label.push_str(&format!(": {:.decimal_places$}", self.confidence));
} else {
label.push_str(&format!("{:.4}", self.confidence));
label.push_str(&format!("{:.decimal_places$}", self.confidence));
}
}
label

6
src/ys/keypoint.rs
View File

@@ -215,7 +215,7 @@ impl Keypoint {
self.name.as_ref()
}
pub fn label(&self, with_name: bool, with_conf: bool) -> String {
pub fn label(&self, with_name: bool, with_conf: bool, decimal_places: usize) -> String {
let mut label = String::new();
if with_name {
label.push_str(
@@ -228,9 +228,9 @@ impl Keypoint {
}
if with_conf {
if with_name {
label.push_str(&format!(": {:.4}", self.confidence));
label.push_str(&format!(": {:.decimal_places$}", self.confidence));
} else {
label.push_str(&format!("{:.4}", self.confidence));
label.push_str(&format!("{:.decimal_places$}", self.confidence));
}
}
label

6
src/ys/mbr.rs
View File

@@ -104,7 +104,7 @@ impl Mbr {
self.confidence
}
pub fn label(&self, with_name: bool, with_conf: bool) -> String {
pub fn label(&self, with_name: bool, with_conf: bool, decimal_places: usize) -> String {
let mut label = String::new();
if with_name {
label.push_str(
@@ -117,9 +117,9 @@ impl Mbr {
}
if with_conf {
if with_name {
label.push_str(&format!(": {:.4}", self.confidence));
label.push_str(&format!(": {:.decimal_places$}", self.confidence));
} else {
label.push_str(&format!("{:.4}", self.confidence));
label.push_str(&format!("{:.decimal_places$}", self.confidence));
}
}
label

6
src/ys/polygon.rs
View File

@@ -73,7 +73,7 @@ impl Polygon {
self.confidence
}
pub fn label(&self, with_name: bool, with_conf: bool) -> String {
pub fn label(&self, with_name: bool, with_conf: bool, decimal_places: usize) -> String {
let mut label = String::new();
if with_name {
label.push_str(
@@ -86,9 +86,9 @@ impl Polygon {
}
if with_conf {
if with_name {
label.push_str(&format!(": {:.4}", self.confidence));
label.push_str(&format!(": {:.decimal_places$}", self.confidence));
} else {
label.push_str(&format!("{:.4}", self.confidence));
label.push_str(&format!("{:.decimal_places$}", self.confidence));
}
}
label