P y R e t r i 代 码 实 讲 PyRetri代码实讲 PyRetri代码实讲
实例操作教程
工程目录
一 configs
配置文件
二 data_jsons
生成gallery和query的json数据
三 main
main函数,虽然main是主函数,但是主要代码还是在pyretri中
四 pyretri
主要代码
五 search
六 data_eda.ipynb
数据分析
七 finetune.py
模型微调
图像特征提取器
八 gen_ret.py
生成结果
补充:通过修改config文件可以得到不同的效果
九 gen_split_file.ipynb
改数据格式
补充:将自己的数据格式,改成pyretri用的开源的数据格式
变成数据格式如下:
0表示query
1表示gallery
最终结果存储在****.txt
中
十 index_result_info_finetune.npy
抽取结果文件
十一 submit.ipynb
top-n提取