修改跟踪算法cpu gpu切换错误

2025-01-15 15:38:50 +08:00
parent 31db54fa37
commit ad850221c5
4 changed files with 16 additions and 11 deletions
--- a/tracking/dotrack/dotracks_back.py
+++ b/tracking/dotrack/dotracks_back.py
@ -153,8 +153,8 @@ class doBackTracks(doTracks):
        
        hand_ious = []
        
-        hboxes = np.empty(shape=(0, 9), dtype = np.float)
-        gboxes = np.empty(shape=(0, 9), dtype = np.float)
+        hboxes = np.empty(shape=(0, 9), dtype = float)
+        gboxes = np.empty(shape=(0, 9), dtype = float)

        
        # start, end 为索引值，需要 start:(end+1)