mirror of
https://github.com/mii443/usls.git
synced 2025-08-22 15:45:41 +00:00
Enhance Annotator with mask-cutout functionality (#101)
Co-authored-by: jamjamjon <zhangjian@zhuofansoft.com>
This commit is contained in:
@ -43,7 +43,8 @@ fn main() -> anyhow::Result<()> {
|
||||
let ys = model.forward(&xs)?;
|
||||
|
||||
// annotate
|
||||
let annotator = Annotator::default();
|
||||
let annotator =
|
||||
Annotator::default().with_mask_style(usls::Style::mask().with_mask_cutout(true));
|
||||
for (x, y) in xs.iter().zip(ys.iter()) {
|
||||
annotator.annotate(x, y)?.save(format!(
|
||||
"{}.jpg",
|
||||
|
Reference in New Issue
Block a user