lichen/detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dotrack haved modified!

Browse Source
This commit is contained in:
王庆刚
2024-05-29 20:31:28 +08:00
parent dff029de20
commit d1ea304491
19 changed files with 379 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tracking/test_tracking.py
View File

@ -21,7 +21,7 @@ from pathlib import Path
# ================= using for import ultralytics
import sys
sys.path.append(r"D:\yolov5track")
sys.path.append(r"D:\DetectTracking")
from utils.gen import Profile
from dotrack.dotracks_back import doBackTracks
@ -68,7 +68,7 @@ def detect_start_end(bboxes, features_dict, filename):
return start, end
def save_subimgs(vts, file):
def save_subimgs(vts, file, TracksDict):
imgdir = Path(f'./result/imgs/{file}')
if not imgdir.exists():
imgdir.mkdir(parents=True, exist_ok=True)
@ -78,19 +78,21 @@ def save_subimgs(vts, file):
for ii in range(len(boxes)):
tid, fid, bid = int(boxes[ii, 4]), int(boxes[ii, 7]), int(boxes[ii, 8])
img = vts.TracksDict[f"frame_{fid}"]["imgs"][bid]
img = TracksDict[f"frame_{fid}"]["imgs"][bid]
# feat = TracksDict[f"frame_{fid}"]["feats"][bid]
# box = TracksDict[f"frame_{fid}"]["boxes"][bid]
cv2.imwrite(str(imgdir) + f"/{tid}_{fid}_{bid}.png", img)
def have_tracked():
trackdict = r'./data/trackdicts'
trackdict = r'./data/trackdicts_1'
alltracks = []
k = 0
gt = Profile()
for filename in os.listdir(trackdict):
filename = 'test_20240402-173935_6920152400975_back_174037372.pkl'
# filename = 'test_20240402-173935_6920152400975_back_174037372.pkl'
filename = '加购_91.pkl'
file, ext = os.path.splitext(filename)
filepath = os.path.join(trackdict, filename)
TracksDict = np.load(filepath, allow_pickle=True)
@ -101,7 +103,7 @@ def have_tracked():
vts = doFrontTracks(bboxes, TracksDict)
vts.classify()
save_subimgs(vts, file)
save_subimgs(vts, file, TracksDict)
plt = plot_frameID_y2(vts)
@ -114,7 +116,7 @@ def have_tracked():
vts.classify()
alltracks.append(vts)
save_subimgs(vts, file)
save_subimgs(vts, file, TracksDict)
edgeline = cv2.imread("./shopcart/cart_tempt/edgeline.png")
draw_all_trajectories(vts, edgeline, save_dir, filename)
print(file+f" need time: {gt.dt:.2f}s")