This commit is contained in:
王庆刚
2025-03-13 15:36:29 +08:00
parent 0efe8892f3
commit 9b5b135fa3
21 changed files with 837 additions and 258 deletions

View File

@ -17,6 +17,7 @@ import warnings
import sys
sys.path.append(r"D:\DetectTracking")
from tracking.utils.read_data import extract_data_realtime, read_tracking_output_realtime
from tracking.utils.plotting import Annotator, colors, draw_tracking_boxes
from tracking.utils import Boxes, IterableSimpleNamespace, yaml_load
@ -85,6 +86,12 @@ def tracking(bboxes, ffeats):
return TrackBoxes, TracksDict
def read_imgs(imgspath, CamerType):
'''
inputs:
@ -440,6 +447,7 @@ def main():
if __name__ == "__main__":
# main_loop()
main()
# try:

View File

@ -330,6 +330,7 @@ def read_similar(filePath):
SimiDict['one2one'] = []
SimiDict['one2SN'] = []
SimiDict['one2n'] = []
SimiDict['algroStartToEnd'] = -1
with open(filePath, 'r', encoding='utf-8') as f:
lines = f.readlines()
@ -352,6 +353,13 @@ def read_similar(filePath):
one2one_list, one2SN_list, one2n_list = [], [], []
Flag_1to1, Flag_1toSN, Flag_1ton = False, False, False
continue
if line.find("algroStartToEnd")>=0:
alg_during = line.split(':')[1].strip()
SimiDict['algroStartToEnd'] = int(alg_during)
if line.find('oneToOne')>=0:
Flag_1to1, Flag_1toSN, Flag_1ton = True, False,False
@ -387,7 +395,7 @@ def read_similar(filePath):
Dict['barcode'] = ''
if label.find("_") > 0:
bcd = label.split('_')[-1]
if len(bcd)>=10 and bcd.isdigit():
if len(bcd)>=8 and bcd.isdigit():
Dict['barcode'] = bcd