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

modify pipeline.py

Browse Source
This commit is contained in:
王庆刚
2025-01-13 17:35:15 +08:00
parent 3d13b0d9c5
commit a50f777839
12 changed files with 322 additions and 211 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
contrast/one2one_contrast.py
View File

@ -194,6 +194,7 @@ def simi_calc(event, stdfeat):
if len(evtfeat)==0 or len(stdfeat)==0:
return None, None, None
evtfeat /= np.linalg.norm(evtfeat, axis=1)[:, None]
stdfeat /= np.linalg.norm(stdfeat, axis=1)[:, None]
@ -305,8 +306,8 @@ def one2SN_pr(evtList, evtDict, stdDict):
barcodes, similars = [], []
for stdbcd in bcd_selected:
stdfeat = stdDict[stdbcd]
# simi_mean, simi_max, simi_mfeat = simi_calc(event, stdfeat)
simi_mean = calc_simil(event, stdfeat)
simi_mean, simi_max, simi_mfeat = simi_calc(event, stdfeat)
# simi_mean = calc_simil(event, stdfeat)
## 在event.front_feats和event.back_feats同时为空时，此处不需要保护
# if simi_mean==None:
@ -629,10 +630,8 @@ if __name__ == '__main__':
stdBarcodePath = r"D:\全实时\source_data\bcdpath"
stdFeaturePath = r"D:\全实时\source_data\stdfeats"
eventSourcePath = [r"D:\全实时\result\pipeline\pipeline"]
resultPath = r"D:\全实时\result\pipeline"
eventSourcePath = [r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\result\ShoppingDict_pkfile"]
resultPath = r"\\192.168.1.28\share\测试视频数据以及日志\算法全流程测试\202412\result\contrast"
eventDataPath = os.path.join(resultPath, "evtobjs")
similPath = os.path.join(resultPath, "simidata")
if not os.path.exists(eventDataPath):
@ -640,7 +639,7 @@ if __name__ == '__main__':
if not os.path.exists(similPath):
os.makedirs(similPath)
test_one2one()
# test_one2one()
test_one2SN()