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

41
tracking/utils/drawtracks.py
View File

@ -87,20 +87,29 @@ def draw_all_trajectories(vts, edgeline, save_dir, filename):
# edgeline2 = edgeline1.copy()
# edgeline = np.concatenate((edgeline1, edgeline2), exis = 1)
# =============================================================================
# '''1. tracks 5点轨迹'''
# trackpth = save_dir.parent /Path("trajectory")/ Path(f"{file}")
# if not trackpth.exists():
# trackpth.mkdir(parents=True, exist_ok=True)
# for track in vts.tracks:
# # if track.cls != 0:
# img = edgeline.copy()
# img = draw5points(track, img)
#
# pth = trackpth.joinpath(f"{track.tid}.png")
# cv2.imwrite(str(pth), img)
#
# =============================================================================
'''1. tracks 5点轨迹'''
trackpth = save_dir.parent /Path("trajectory")/ Path(f"{file}")
if not trackpth.exists():
trackpth.mkdir(parents=True, exist_ok=True)
for track in vts.tracks:
# if track.cls != 0:
img = edgeline.copy()
img = draw5points(track, img)
pth = trackpth.joinpath(f"{track.tid}.png")
cv2.imwrite(str(pth), img)
for track in vts.merged_tracks:
# if track.cls != 0:
img = edgeline.copy()
img = draw5points(track, img)
pth = trackpth.joinpath(f"{track.tid}_.png")
cv2.imwrite(str(pth), img)
'''2. all tracks 中心轨迹'''
img1, img2 = edgeline.copy(), edgeline.copy()
@ -139,7 +148,7 @@ def draw_all_trajectories(vts, edgeline, save_dir, filename):
def drawFeatures(allvts, save_dir):
# [trajlen_min, trajdist_max, trajlen_rate, trajist_rate]]
feats = [track.feature for vts in allvts for track in vts.tracks]
feats = [track.TrajFeat for vts in allvts for track in vts.tracks]
feats = np.array(feats)
fig, ax = plt.subplots()
ax.scatter(feats[:,3], feats[:, 1], s=10)
@ -269,7 +278,7 @@ def draw5points(track, img):
trajstd = 0
trajlen_min, trajlen_max, trajdist_min, trajdist_max, trajlen_rate, trajdist_rate = track.feature
trajlen_min, trajlen_max, trajdist_min, trajdist_max, trajlen_rate, trajdist_rate = track.TrajFeat
for i in range(boxes.shape[0]):
cv2.circle(img, (int(cornpoints[i, 0]), int(cornpoints[i, 1])), 6, (255, 255, 255), 2)