PyRetri代码实讲

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提取


  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值