From 7f9895d17cc8113dc5c6b245370d3ae88c4f2242 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ga=C3=ABl=20Cartier-Michaud?= <gael.cartierm@hes-so.ch>
Date: Sat, 2 Sep 2023 16:38:32 +0200
Subject: [PATCH] modif no harvest mode
---
demo/detector.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/demo/detector.py b/demo/detector.py
index 6b0563ae..e06d4150 100644
--- a/demo/detector.py
+++ b/demo/detector.py
@@ -16,8 +16,8 @@ class Detector :
self.nothing_counter = 0
self.myCam = Cam( config )
self.myModel = Model( config )
- self.myArm = Arm( config )
if self.harvest :
+ self.myArm = Arm( config )
self.myModel.add_callback( self.__harvestCallback )
self.harvest_job = Thread( target = self.myArm.harvest, args = ( self.myArm.target, ) )
@@ -59,8 +59,8 @@ class Detector :
for img in self.myCam.stream() :
self.myModel.predict( img )
if cv2.waitKey(25) & 0xFF == ord('q') or self.exit_status :
- self.myArm.closeCommunication()
cv2.destroyAllWindows()
+ if self.harvest : self.myArm.closeCommunication()
break
--
GitLab