@@ 73-84 (lines=12) @@ | ||
70 | # use the first file, nq should be less than VECTORS_PER_FILE |
|
71 | if nq > MAX_NQ: |
|
72 | raise Exception("Over size nq") |
|
73 | if data_type == "random": |
|
74 | file_name = SRC_BINARY_DATA_DIR+'query_%d.npy' % dimension |
|
75 | elif data_type == "sift": |
|
76 | file_name = SIFT_SRC_DATA_DIR+'query.npy' |
|
77 | elif data_type == "deep": |
|
78 | file_name = DEEP_SRC_DATA_DIR+'query.npy' |
|
79 | elif data_type == "jaccard": |
|
80 | file_name = JACCARD_SRC_DATA_DIR+'query.npy' |
|
81 | elif data_type == "hamming": |
|
82 | file_name = HAMMING_SRC_DATA_DIR+'query.npy' |
|
83 | elif data_type == "sub" or data_type == "super": |
|
84 | file_name = STRUCTURE_SRC_DATA_DIR+'query.npy' |
|
85 | data = np.load(file_name) |
|
86 | vectors = data[0:nq].tolist() |
|
87 | return vectors |
|
@@ 54-65 (lines=12) @@ | ||
51 | def gen_file_name(idx, dimension, data_type): |
|
52 | s = "%05d" % idx |
|
53 | fname = FILE_PREFIX + str(dimension) + "d_" + s + ".npy" |
|
54 | if data_type == "random": |
|
55 | fname = SRC_BINARY_DATA_DIR+fname |
|
56 | elif data_type == "sift": |
|
57 | fname = SIFT_SRC_DATA_DIR+fname |
|
58 | elif data_type == "deep": |
|
59 | fname = DEEP_SRC_DATA_DIR+fname |
|
60 | elif data_type == "jaccard": |
|
61 | fname = JACCARD_SRC_DATA_DIR+fname |
|
62 | elif data_type == "hamming": |
|
63 | fname = HAMMING_SRC_DATA_DIR+fname |
|
64 | elif data_type == "sub" or data_type == "super": |
|
65 | fname = STRUCTURE_SRC_DATA_DIR+fname |
|
66 | return fname |
|
67 | ||
68 |