- updated spindle sequence

- improved marker settings
- minimized number of open windows

ocv_corner_tracker.py

@@ -84,6 +84,8 @@ class Corner:
 		return matcher_bbox_local, matcher_res
 
 	def path_add(self, point):
+		if len(self.path) > 200:
+			return
 		if self.line_from is not None:
 			self._path.append({'from': self.line_from, 'to': point})
 		self.line_from = point
@@ -127,18 +129,27 @@ class CornerTracker:
 		tracker_rect = to_rect(self.tracking_bb)
 		cv2.rectangle(image_anno, tracker_rect[0], tracker_rect[1], self.color, 1)
 
-	def init_reference_frame(self, image: np.array, select_window: np.array):
+	def init_reference_frame(self, image: np.array):
 		print(f"Select tracking object")
-		self.tracking_ref_bb = cv2.selectROI("Tracker Reference", select_window, False)
-		self.tracking_ref_img = image_crop(image.copy(), self.tracking_ref_bb)
-		self.tracking_ref_gray_img = cv2.cvtColor(self.tracking_ref_img, cv2.COLOR_BGR2GRAY)
+		bb = cv2.selectROI("Tracker Reference", image, False)
+		cv2.destroyWindow("Tracker Reference")
+		if bbox_center(bb) == (0, 0):
+			return False
 
+		self.tracking_ref_bb = bb
+		self.tracking_ref_img = image_crop(image, self.tracking_ref_bb)
+		self.tracking_ref_gray_img = cv2.cvtColor(self.tracking_ref_img, cv2.COLOR_BGR2GRAY)
 		self.corner_matcher_list = []
+
 		corner_list = []
 		count = 1
 		while True:
 			print(f"Add Corner {count}")
-			self.matching_tpl_bb = cv2.selectROI("Matcher Reference", self.tracking_ref_img, fromCenter=True, showCrosshair=True)
+			bb = cv2.selectROI("Matcher Reference", self.tracking_ref_img, fromCenter=True, showCrosshair=True)
+			if bbox_center(bb) == (0, 0):
+				break
+
+			self.matching_tpl_bb = bb
 			self.matching_tpl_img = image_crop(self.tracking_ref_gray_img.copy(), self.matching_tpl_bb)
 			self.corner_matcher_list.append(Corner(self.matching_tpl_img, self.matching_tpl_bb, name=f"Corner-{count}"))
 			corner_list.append(bbox_center(bbox_add_position(self.matching_tpl_bb, self.tracking_ref_bb)))
@@ -146,11 +157,9 @@ class CornerTracker:
 			print(f"Corner {count} added")
 			print(f"Press any key to add another corner or ESC to continue")
 
-			k = cv2.waitKey(-1) & 0xff
-			if k == 27:
-				break
 			count += 1
 
+		cv2.destroyWindow("Matcher Reference")
 		print(f"Added {count} corners")
 
 		# Initialize tracker with first frame and bounding box
@@ -160,6 +169,8 @@ class CornerTracker:
 		# Refine initial corners and store them as reference
 		self.corner_ref = self._corner_refine(cv2.cvtColor(image, cv2.COLOR_BGR2GRAY), corners=corner_list)
 
+		return len(corner_list) > 0
+
 	def process(self, image: np.array, image_anno: np.array):
 		if self.tracker is None:
 			raise Exception(f"{self.name}: Call init_reference_frame() first")
@@ -260,19 +271,18 @@ if __name__ == '__main__':
 	if video.isOpened():
 		# Read first frame.
 		ok, image = video.read()
-		select_window = image.copy()
 		if ok:
 			for tracker_count in range(0, 6):
+				select_window = image.copy()
 				print(f"Add tracker #{tracker_count}")
 				ct = CornerTracker(colors[tracker_count], name=f"Tracker-{tracker_count}")
-				ct.init_reference_frame(image, select_window)
-				tracker_list.append(ct)
-				print(f"Press any key to add another tracker or ESC to continue")
-				k = cv2.waitKey(-1) & 0xff
-				if k == 27:
+				ok = ct.init_reference_frame(select_window)
+				if ok:
+					tracker_list.append(ct)
+					print(f"Number of active tracker: {len(tracker_list)}")
+				else:
 					break
 
-				print(f"Number of active tracker: {len(tracker_list)}")
 		else:
 			print('Cannot read video file')
 
@@ -285,8 +295,8 @@ if __name__ == '__main__':
 				i = 0
 				for ct in tracker_list:
 					mean_distance = ct.process(image, image_anno)
-					cv2.putText(image_anno, f"Distance [{i}] : {mean_distance}", (50, 20 + 20 * i),
-								cv2.FONT_HERSHEY_SIMPLEX, 0.6, ct.color, 1)
+					cv2.putText(image_anno, f"Distance [{i}] : ({mean_distance[0]:+05.2f}, {mean_distance[1]:+05.2f})", (20, 25 + 25 * i),
+								cv2.FONT_HERSHEY_SIMPLEX, 0.5, ct.color, 1)
 					i += 1
 				cv2.imshow(f"Image Anno", image_anno)
 			else: