lichen/ieemoo-ai-contrast
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

智能秤分析

Browse Source
This commit is contained in:
lee
2025-08-06 17:03:28 +08:00
parent 54898e30ec
commit 3392d76e38
17 changed files with 572 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tools/similar_analysis.py
View File

@ -10,7 +10,7 @@ import yaml
import os
class analysis:
class SimilarAnalysis:
def __init__(self):
with open('../configs/similar_analysis.yml', 'r') as f:
self.conf = yaml.load(f, Loader=yaml.FullLoader)
@ -53,6 +53,7 @@ class analysis:
def get_feature_map(self, all_imgs):
feature_dicts = {}
for img_pth in all_imgs:
print(f"Processing {img_pth}")
feature_dict = self.get_feature(img_pth)
feature_dicts = dict(ChainMap(feature_dict, feature_dicts))
return feature_dicts
@ -85,7 +86,7 @@ class analysis:
feature_dict2 = feature_dicts[img_pth2]
similarity = self.get_similarity(feature_dict1.cpu().numpy(),
feature_dict2.cpu().numpy())
dir_name = img_pth1.split(os.sep)[-3]
dir_name = img_pth1.split('/')[-3]
save_path = os.sep.join([self.conf['data']['image_joint_pth'], dir_name])
if similarity > 0.7:
merge_imgs(img_pth1,
@ -99,7 +100,7 @@ class analysis:
if __name__ == '__main__':
ana = analysis()
ana = SimilarAnalysis()
all_imgs = ana.create_total_feature()
feature_dicts = ana.get_feature_map(all_imgs)
all_compare_img = ana.get_image_map()