update

tracking/dotrack/dotracks_back.py

@@ -5,8 +5,15 @@ Created on Mon Mar  4 18:36:31 2024
 @author: ym
 """
 import numpy as np
+import cv2
 from tracking.utils.mergetrack import track_equal_track
 from scipy.spatial.distance import cdist
+from pathlib import Path
+curpath = Path(__file__).resolve().parents[0]
+curpath = Path(curpath)
+parpath = curpath.parent
+
+
 from .dotracks import doTracks, ShoppingCart
 from .track_back import backTrack
 
@@ -18,10 +25,26 @@ class doBackTracks(doTracks):
         
         self.tracks = [backTrack(b, f) for b, f in zip(self.lboxes, self.lfeats)]
         
-        # self.similar_dict = self.similarity()
+        # self.similar_dict = self.similarity()        
+        # self.shopcart = ShoppingCart(bboxes)
+
+        self.incart = self.getincart()
+        
+        
+    def getincart(self):
+        img1 = cv2.imread(str(parpath/'shopcart/cart_tempt/incart.png'), cv2.IMREAD_GRAYSCALE)
+        img2 = cv2.imread(str(parpath/'shopcart/cart_tempt/cartedge.png'), cv2.IMREAD_GRAYSCALE)
+        
+        
+        ret, binary1 = cv2.threshold(img1, 250, 255, cv2.THRESH_BINARY)
+        ret, binary2 = cv2.threshold(img2, 250, 255, cv2.THRESH_BINARY)
+        
+        binary = cv2.bitwise_or(binary1, binary2)
+        
+        
+        return binary
         
         
-        self.shopcart = ShoppingCart(bboxes)
 
     def classify(self):
         '''功能：对 tracks 中元素分类 '''