Skip to content

Commit ece8558

Browse files
author
BIN LI
committedAug 10, 2021
fix a bug for reading multiple mags
1 parent 78ecfb0 commit ece8558

File tree

2 files changed

+1
-0
lines changed

2 files changed

+1
-0
lines changed
 

‎compute_feats.py

+1
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def compute_tree_feats(args, bags_list, embedder_low, embedder_high, save_path=N
9292
feats_list.extend(feats)
9393
for idx, low_patch in enumerate(low_patches):
9494
high_patches = glob.glob(low_patch.replace('.jpeg', os.sep+'*.jpg')) + glob.glob(low_patch.replace('.jpeg', os.sep+'*.jpeg'))
95+
high_patches = high_patches + glob.glob(low_patch.replace('.jpg', os.sep+'*.jpg')) + glob.glob(low_patch.replace('.jpg', os.sep+'*.jpeg'))
9596
if len(high_patches) == 0:
9697
pass
9798
else:

‎init.pth

-5.22 KB
Binary file not shown.

0 commit comments

Comments
 (0)
Please sign in to comment.