diff --git a/src/core/annotator.rs b/src/core/annotator.rs index 694cb7e..9d07cc1 100644 --- a/src/core/annotator.rs +++ b/src/core/annotator.rs @@ -115,7 +115,51 @@ impl Annotator { pub fn new() -> Self { Default::default() } - + pub fn new_with_fonts(font: &str) -> Self { + Self { + font: match Self::load_font(font) { + Ok(x) => x, + Err(err) => panic!("Failed to load font: {}", err), + }, + _scale: 6.666667, + scale_dy: 28., + polygons_alpha: 179, + saveout: None, + saveout_subs: vec![], + saveout_base: String::from("runs"), + decimal_places: 4, + without_bboxes: false, + without_bboxes_conf: false, + without_bboxes_name: false, + bboxes_text_color: Rgba([0, 0, 0, 255]), + bboxes_thickness: 1, + bboxes_thickness_threshold: 0.3, + without_bboxes_text_bg: false, + without_mbrs: false, + without_mbrs_conf: false, + without_mbrs_name: false, + without_mbrs_text_bg: false, + mbrs_text_color: Rgba([0, 0, 0, 255]), + without_keypoints: false, + with_keypoints_conf: false, + with_keypoints_name: false, + keypoints_radius: 3, + skeletons: None, + keypoints_palette: None, + without_keypoints_text_bg: false, + keypoints_text_color: Rgba([0, 0, 0, 255]), + without_polygons: false, + without_contours: false, + contours_color: Rgba([255, 255, 255, 255]), + with_polygons_name: false, + with_polygons_conf: false, + with_polygons_text_bg: false, + polygons_text_color: Rgba([255, 255, 255, 255]), + probs_topk: 5usize, + without_masks: false, + colormap: None, + } + } pub fn with_decimal_places(mut self, x: usize) -> Self { self.decimal_places = x; self