From f2c9eac332bf62a0cf69bbbc1dcb2f2d6e794171 Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Tue, 17 Mar 2020 23:33:33 -0400
Subject: [PATCH] [robots] (demo) Add a print of target
---
robots-at-robots/research/demos/demo_pipeline.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/robots-at-robots/research/demos/demo_pipeline.py b/robots-at-robots/research/demos/demo_pipeline.py
index af54491..7a1e2f6 100644
--- a/robots-at-robots/research/demos/demo_pipeline.py
+++ b/robots-at-robots/research/demos/demo_pipeline.py
@@ -1,5 +1,6 @@
import cv2
+from polystar.common.communication.print_target_sender import PrintTargetSender
from polystar.common.models.camera import Camera
from polystar.common.models.label_map import LabelMap
from polystar.common.models.object import ObjectType
@@ -28,6 +29,7 @@ if __name__ == "__main__":
objects_validators=[ConfidenceObjectValidator(0.6), TypeObjectValidator(ObjectType.Armor)],
object_selector=ClosestObjectSelector(),
target_factory=RatioSimpleTargetFactory(injector.get(Camera), 300, 100),
+ target_sender=PrintTargetSender(),
)
with PltResultViewer("Demo of tf model") as viewer:
--
GitLab