update for bakeup

2024-07-22 20:16:45 +08:00
parent e986ec060b
commit 16543107f3
16 changed files with 179 additions and 52 deletions
--- a/tracking/utils/drawtracks.py
+++ b/tracking/utils/drawtracks.py
@ -98,11 +98,12 @@ def draw_all_trajectories(vts, edgeline, save_dir, file, draw5p=False):
    H, W = img.shape[:2]
    cv2.line(img, (int(W/2), 0), (int(W/2), H), (128, 255, 128), 2)
    
-    imgpth = save_dir.joinpath(f"{file}_show.png")
-    cv2.imwrite(str(imgpth), img)
+    
+    # imgpth = save_dir.joinpath(f"{file}_show.png")
+    # cv2.imwrite(str(imgpth), img)

    if not draw5p:
-        return
+        return img
        
    ''' tracks 5点轨迹'''
    trackpth = save_dir / Path("trajectory") / Path(f"{file}")