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

fix some nots

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