submit.ipynb

$$submit.ipynb$$

import pickle
import numpy as np
import pandas as pd
import os
from matplotlib import pyplot as plt

file_path1 = '/data/nextcloud/dbc2017/files/jupyter/PyRetri/data_jsons/caltech_gallery.json'
file_path2 = '/data/nextcloud/dbc2017/files/jupyter/PyRetri/index_result_info_finetune.npy'

with open(file_path1,'rb') as f:
    caltech_gallery = pickle.load(f)
index_result_info = np.load(file_path2,allow_pickle=True)

topk = 1
index = 1
topk_idx = index_result_info[index]['ranked_neighbors_idx'][:topk]
query_name = os.path.basename(index_result_info[index]['path'])
gallery_name = os.path.basename(caltech_gallery['info_dicts'][topk_idx[0]]['path'])

print(gallery_name)
print(query_name)

plt.imshow(plt.imread(os.path.join('/data/caltech101/query/allimg/',query_name)))

plt.imshow(plt.imread(os.path.join('/data/caltech101/gallery/allimg/',gallery_name)))

# 打印前 10 的数据
topk = 10
result = ''
for index in range(len(index_result_info)):
    topk_idx = index_result_info[index]['ranked_neighbors_idx'][:topk]
    query_name = os.path.basename(index_result_info[index]['path'])
    line = query_name + ',{'
    for i in range(10):
        gallery_name = os.path.basename(caltech_gallery['info_dicts'][topk_idx[i]]['path'])
        line+=gallery_name+','
    line = line[:-1]+'}\n'
    result += line

# 写入提交文件
with open('submission.csv', 'a') as f:
    f.truncate(0)
    f.write(result)
    f.close()