update

tracking/dotrack/dotracks_front.py

@@ -4,7 +4,13 @@ Created on Mon Mar  4 18:38:20 2024
 
 @author: ym
 """
+import cv2
 import numpy as np
+from pathlib import Path
+
+curpath = Path(__file__).resolve().parents[0]
+curpath = Path(curpath)
+parpath = curpath.parent
 # from tracking.utils.mergetrack import track_equal_track
 from .dotracks import doTracks
 from .track_front import frontTrack
@@ -16,6 +22,14 @@ class doFrontTracks(doTracks):
         # self.tracks = [frontTrack(b) for b in self.lboxes]
         self.tracks = [frontTrack(b, f) for b, f in zip(self.lboxes, self.lfeats)]
         
+        self.incart = self.getincart()
+        
+    def getincart(self):
+        img = cv2.imread(str(parpath/'shopcart/cart_tempt/incart_ftmp.png'), cv2.IMREAD_GRAYSCALE)
+        ret, binary = cv2.threshold(img, 250, 255, cv2.THRESH_BINARY)
+        
+        return binary
+            
 
     def classify(self):
         '''功能：对 tracks 中元素分类 '''