This commit is contained in:
王庆刚
2024-11-25 18:05:08 +08:00
parent c47894ddc0
commit 8bbee310ba
109 changed files with 1003 additions and 305 deletions

View File

@ -49,7 +49,8 @@ from datetime import datetime
sys.path.append(r"D:\DetectTracking")
from tracking.utils.read_data import extract_data, read_tracking_output, read_one2one_simi, read_deletedBarcode_file
from genfeats import genfeatures, stdfeat_infer
from config import config as conf
from genfeats import model_init, genfeatures, stdfeat_infer
IMG_FORMAT = ['.bmp', '.jpg', '.jpeg', '.png']
@ -546,10 +547,10 @@ def test_one2one():
model = model_init(conf)
'''==== 1. 生成标准特征集, 只需运行一次 ==============='''
genfeatures(stdSamplePath, stdBarcodePath, stdFeaturePath, bcdSet)
genfeatures(model, stdSamplePath, stdBarcodePath, stdFeaturePath, bcdSet)
print("stdFeats have generated and saved!")