mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Update predict_system.py

Browse Source
This commit is contained in:
Bin Lu 2021-06-07 15:51:06 +08:00 committed by GitHub
parent 75b9b6d284
commit d3972cf5a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deploy/python/predict_system.py
View File

@ -33,7 +33,7 @@ from utils.get_image_list import get_image_list
def split_datafile(data_file, image_root): def split_datafile(data_file, image_root):
gallery_images = [] gallery_images = []
gallery_docs = [] gallery_docs = []
with open(datafile) as f: with open(data_file) as f:
lines = f.readlines() lines = f.readlines()
for i, line in enumerate(lines): for i, line in enumerate(lines):
line = line.strip().split("\t") line = line.strip().split("\t")
@ -60,7 +60,7 @@ class SystemPredictor(object):
def indexer(self, config): def indexer(self, config):
if 'build' in config.keys() and config['build']['enable']: # build the index from scratch if 'build' in config.keys() and config['build']['enable']: # build the index from scratch
with open(config['build']['datafile']) as f: with open(config['build']['data_file']) as f:
lines = f.readlines() lines = f.readlines()
gallery_images, gallery_docs = split_datafile(config['build']['data_file'], config['build']['image_root']) gallery_images, gallery_docs = split_datafile(config['build']['data_file'], config['build']['image_root'])
# extract gallery features # extract gallery features
@ -85,7 +85,7 @@ class SystemPredictor(object):
xmin, ymin, xmax, ymax = result["bbox"].astype("int") xmin, ymin, xmax, ymax = result["bbox"].astype("int")
crop_img = img[xmin:xmax, ymin:ymax, :].copy() crop_img = img[xmin:xmax, ymin:ymax, :].copy()
rec_results = self.rec_predictor.predict(crop_img) rec_results = self.rec_predictor.predict(crop_img)
result["featrue"] = rec_results result["feature"] = rec_results
scores, docs = self.Searcher.search(query=rec_results, return_k=self.return_k, search_budget=self.search_budget) scores, docs = self.Searcher.search(query=rec_results, return_k=self.return_k, search_budget=self.search_budget)
result["ret_docs"] = docs result["ret_docs"] = docs