提交 b05c8b1b 编写于 作者: H HydrogenSulfate 提交者: Walter

fix(retrieval): remove max_iter in retrieval.py to avoid empty data

上级 62c9d0f4
......@@ -125,12 +125,8 @@ def compute_feature(engine, name="gallery"):
all_feat = []
all_label = []
all_camera = []
max_iter = len(dataloader) - 1 if platform.system() == "Windows" else len(
dataloader)
has_camera = False
for idx, batch in enumerate(dataloader): # load is very time-consuming
if idx >= max_iter:
break
if idx % engine.config["Global"]["print_batch_step"] == 0:
logger.info(
f"{name} feature calculation process: [{idx}/{len(dataloader)}]"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册