未验证 提交 5db7d2d4 编写于 作者: B Bin Lu 提交者: GitHub

Update build_gallery.py

上级 cf622ea0
...@@ -69,7 +69,7 @@ class GalleryBuilder(object): ...@@ -69,7 +69,7 @@ class GalleryBuilder(object):
# when remove data in index, do not need extract fatures # when remove data in index, do not need extract fatures
if operation_method != "remove": if operation_method != "remove":
gallery_features = self._extract_features(gallery_images, config) #76 * 512 gallery_features = self._extract_features(gallery_images, config)
assert operation_method in [ assert operation_method in [
"new", "remove", "append" "new", "remove", "append"
...@@ -129,7 +129,7 @@ class GalleryBuilder(object): ...@@ -129,7 +129,7 @@ class GalleryBuilder(object):
# calculate id for new data # calculate id for new data
start_id = max(ids.keys()) + 1 if ids else 0 start_id = max(ids.keys()) + 1 if ids else 0
ids_now = ( ids_now = (
np.arange(0, len(gallery_images)) + start_id).astype(np.int64) #ids: just the number sequence np.arange(0, len(gallery_images)) + start_id).astype(np.int64)
# only train when new index file # only train when new index file
if operation_method == "new": if operation_method == "new":
...@@ -189,7 +189,7 @@ class GalleryBuilder(object): ...@@ -189,7 +189,7 @@ class GalleryBuilder(object):
batch_img.append(img) batch_img.append(img)
if (i + 1) % batch_size == 0: if (i + 1) % batch_size == 0:
rec_feat = self.rec_predictor.predict(batch_img) #32 * 512 rec_feat = self.rec_predictor.predict(batch_img)
gallery_features[i - batch_size + 1:i + 1, :] = rec_feat gallery_features[i - batch_size + 1:i + 1, :] = rec_feat
batch_img = [] batch_img = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册