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

guoqing bakeup

Browse Source
This commit is contained in:
王庆刚
2024-10-04 12:12:44 +08:00
parent 09e92d63b3
commit 390c5d2d94
37 changed files with 1409 additions and 219 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tracking/tracking_test.py
View File

@ -80,14 +80,14 @@ def save_subimgs(vts, file, TracksDict):
cv2.imwrite(str(imgdir) + f"/{tid}_{fid}_{bid}.png", img)
def have_tracked():
trackdict = r'./data/trackdicts_20240608'
trackdict = r'./data/trackdicts'
alltracks = []
k = 0
gt = Profile()
for filename in os.listdir(trackdict):
# filename = 'test_20240402-173935_6920152400975_back_174037372.pkl'
filename = '6907149227609_20240508-174733_back_returnGood_70f754088050_425_17327712807.pkl'
filename = '6907149227609_20240508-174733_front_returnGood_70f754088050_425_17327712807.pkl'
# filename = '6907149227609_20240508-174733_back_returnGood_70f754088050_425_17327712807.pkl'
# filename = '6907149227609_20240508-174733_front_returnGood_70f754088050_425_17327712807.pkl'
file, ext = os.path.splitext(filename)
filepath = os.path.join(trackdict, filename)
@ -119,11 +119,14 @@ def have_tracked():
save_subimgs(vts, file, TracksDict)
edgeline = cv2.imread("./shopcart/cart_tempt/edgeline.png")
img_tracking = draw_all_trajectories(vts, edgeline, save_dir, file)
trackpath = save_dir.joinpath(f'{file}.png')
cv2.imwrite(str(trackpath), img_tracking)
print(file+f" need time: {gt.dt:.2f}s")
k += 1
if k==1:
break
# k += 1
# if k==1:
# break
if len(alltracks):
drawFeatures(alltracks, save_dir)