modify dotrack module
This commit is contained in:
@ -16,9 +16,6 @@ import pandas as pd
|
||||
from scipy.spatial.distance import cdist
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
|
||||
# ================= using for import ultralytics
|
||||
import sys
|
||||
sys.path.append(r"D:\DetectTracking")
|
||||
@ -85,13 +82,13 @@ def save_subimgs(vts, file, TracksDict):
|
||||
cv2.imwrite(str(imgdir) + f"/{tid}_{fid}_{bid}.png", img)
|
||||
|
||||
def have_tracked():
|
||||
trackdict = r'./data/trackdicts_1'
|
||||
trackdict = r'./data/trackdicts'
|
||||
alltracks = []
|
||||
k = 0
|
||||
gt = Profile()
|
||||
for filename in os.listdir(trackdict):
|
||||
# filename = 'test_20240402-173935_6920152400975_back_174037372.pkl'
|
||||
filename = '加购_91.pkl'
|
||||
# filename = '加购_91.pkl'
|
||||
|
||||
file, ext = os.path.splitext(filename)
|
||||
filepath = os.path.join(trackdict, filename)
|
||||
@ -123,9 +120,9 @@ def have_tracked():
|
||||
|
||||
|
||||
|
||||
k += 1
|
||||
if k==1:
|
||||
break
|
||||
# k += 1
|
||||
# if k==1:
|
||||
# break
|
||||
|
||||
if len(alltracks):
|
||||
drawFeatures(alltracks, save_dir)
|
||||
|
Reference in New Issue
Block a user