guoqing bakeup

tracking/dotrack/dotracks.py

@@ -8,10 +8,11 @@ import numpy as np
 import cv2
 from pathlib import Path
 from scipy.spatial.distance import cdist
-from utils.mergetrack import track_equal_track, readDict
+from tracking.utils.mergetrack import track_equal_track, readDict
 curpath = Path(__file__).resolve().parents[0]
 
 curpath = Path(curpath)
+parpath = curpath.parent
 
 class MoveState:
     """商品运动状态标志"""
@@ -297,11 +298,15 @@ class Track:
                    front, 前置摄像头
         '''
         if camerType=="back":
-            incart = cv2.imread("./shopcart/cart_tempt/incart.png", cv2.IMREAD_GRAYSCALE)
-            outcart = cv2.imread("./shopcart/cart_tempt/outcart.png", cv2.IMREAD_GRAYSCALE)
+            incart = cv2.imread(str(parpath/'shopcart/cart_tempt/incart.png'), cv2.IMREAD_GRAYSCALE)
+            outcart = cv2.imread(str(parpath/'shopcart/cart_tempt/outcart.png'), cv2.IMREAD_GRAYSCALE)
         else:
-            incart = cv2.imread("./shopcart/cart_tempt/incart_ftmp.png", cv2.IMREAD_GRAYSCALE)
-            outcart = cv2.imread("./shopcart/cart_tempt/outcart_ftmp.png", cv2.IMREAD_GRAYSCALE)
+            incart = cv2.imread(str(parpath/'shopcart/cart_tempt/incart_ftmp.png'), cv2.IMREAD_GRAYSCALE)
+            outcart = cv2.imread(str(parpath/'shopcart/cart_tempt/outcart_ftmp.png'), cv2.IMREAD_GRAYSCALE)
+            
+            # incart = cv2.imread('./cart_tempt/incart_ftmp.png', cv2.IMREAD_GRAYSCALE)
+            # outcart = cv2.imread('./cart_tempt/outcart_ftmp.png', cv2.IMREAD_GRAYSCALE)
+            
 
         xc, yc = self.cornpoints[:,0].clip(0,self.imgshape[0]-1).astype(np.int64), self.cornpoints[:,1].clip(0,self.imgshape[1]-1).astype(np.int64)
         x1, y1 = self.cornpoints[:,6].clip(0,self.imgshape[0]-1).astype(np.int64), self.cornpoints[:,7].clip(0,self.imgshape[1]-1).astype(np.int64)

tracking/dotrack/dotracks_back.py

@@ -5,7 +5,7 @@ Created on Mon Mar  4 18:36:31 2024
 @author: ym
 """
 import numpy as np
-from utils.mergetrack import track_equal_track
+from tracking.utils.mergetrack import track_equal_track
 from scipy.spatial.distance import cdist
 from .dotracks import doTracks, ShoppingCart
 from .track_back import backTrack

tracking/dotrack/dotracks_front.py

@@ -5,7 +5,7 @@ Created on Mon Mar  4 18:38:20 2024
 @author: ym
 """
 import numpy as np
-from utils.mergetrack import track_equal_track
+# from tracking.utils.mergetrack import track_equal_track
 from .dotracks import doTracks
 from .track_front import frontTrack
 

tracking/dotrack/track_back.py

@@ -10,6 +10,10 @@ from scipy.spatial.distance import cdist
 from sklearn.decomposition import PCA
 from .dotracks import MoveState, Track
 
+from pathlib import Path
+curpath = Path(__file__).resolve().parents[0]
+curpath = Path(curpath)
+parpath = curpath.parent
 
 class backTrack(Track):
     # boxes: [x1, y1, x2, y2, track_id, score, cls, frame_index, box_index]
@@ -93,9 +97,9 @@ class backTrack(Track):
             maxbox_iou, minbox_iou：track中最大、最小 box 和boxes流的iou，二者差值越小，越接近 1，表明track的运动型越小。
             incartrates: 各box和incart的iou时序，由小变大，反应的是置入过程，由大变小，反应的是取出过程
         '''
-        incart = cv2.imread("./shopcart/cart_tempt/incart.png", cv2.IMREAD_GRAYSCALE)
-        outcart = cv2.imread("./shopcart/cart_tempt/outcart.png", cv2.IMREAD_GRAYSCALE)
-        cartboarder = cv2.imread("./shopcart/cart_tempt/cartboarder.png", cv2.IMREAD_GRAYSCALE)
+        incart = cv2.imread(str(parpath/"shopcart/cart_tempt/incart.png"), cv2.IMREAD_GRAYSCALE)
+        outcart = cv2.imread(str(parpath/"shopcart/cart_tempt/outcart.png"), cv2.IMREAD_GRAYSCALE)
+        cartboarder = cv2.imread(str(parpath/"shopcart/cart_tempt/cartboarder.png"), cv2.IMREAD_GRAYSCALE)
         
         incartrates = []
         temp = np.zeros(incart.shape, np.uint8)