mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-08-22 15:45:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor tweaks (#100)

Browse Source
This commit is contained in:
Jamjamjon
2025-05-21 16:58:02 +08:00
committed by GitHub
parent 9826fac208
commit ea89b914eb
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/viz/drawable/mod.rs
View File

@ -61,25 +61,25 @@ pub trait Drawable {
impl Drawable for Y {
fn draw(&self, ctx: &DrawContext, canvas: &mut image::RgbaImage) -> anyhow::Result<()> {
if let Some(probs) = self.probs() {
probs.to_vec().draw(ctx, canvas)?;
probs.draw(ctx, canvas)?;
}
if let Some(polygons) = self.polygons() {
polygons.iter().try_for_each(|x| x.draw(ctx, canvas))?;
polygons.draw(ctx, canvas)?;
}
if let Some(hbbs) = self.hbbs() {
hbbs.iter().try_for_each(|x| x.draw(ctx, canvas))?;
hbbs.draw(ctx, canvas)?;
}
if let Some(obbs) = self.obbs() {
obbs.iter().try_for_each(|x| x.draw(ctx, canvas))?;
obbs.draw(ctx, canvas)?;
}
if let Some(keypoints) = self.keypoints() {
keypoints.to_vec().draw(ctx, canvas)?;
keypoints.draw(ctx, canvas)?;
}
if let Some(keypointss) = self.keypointss() {
keypointss.to_vec().draw(ctx, canvas)?;
keypointss.draw(ctx, canvas)?;
}
if let Some(masks) = self.masks() {
masks.to_vec().draw(ctx, canvas)?;
masks.draw(ctx, canvas)?;
}
Ok(())