提交 1c68b63a 编写于 作者: 悟、's avatar 悟、 提交者: zengshao0622

fix some nots

上级 23b01120
......@@ -162,7 +162,7 @@ DataLoader:
contrast: 0.4
saturation: 0.4
hue: 0.1
p: 1.0
p: 1.0 # refer to official settings
- RandomGrayscale:
p: 0.2
- NormalizeImage:
......
......@@ -162,7 +162,7 @@ DataLoader:
contrast: 0.4
saturation: 0.4
hue: 0.1
p: 1.0
p: 1.0 # refer to official settings
- RandomGrayscale:
p: 0.2
- NormalizeImage:
......
......@@ -15,7 +15,6 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import platform
from collections import defaultdict
import numpy as np
......@@ -257,7 +256,7 @@ def compute_re_ranking_dist(query_feat: paddle.Tensor,
original_dist = np.transpose(original_dist / np.max(original_dist, axis=0))
V = np.zeros_like(original_dist).astype(np.float16)
initial_rank = np.argpartition(original_dist, range(1, k1 + 1)) # 22.2s
initial_rank = np.argpartition(original_dist, range(1, k1 + 1))
logger.info("Start re-ranking...")
for p in range(num_all):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册