1. 预训练模型
下载预模型需要提交在线申请
下载的预训练模型建议存放为../BriVL-pretrain-model/BriVL-1.0-5500w.pth
2. 本实现参考了代码BriVlL推理源码
3. 绝句诗来源于CCPC
已放在data/poetry/CCPC路径下
# a. 安装依赖
pip install -r requirements.txt
# b. 提取诗句特征
python extract_poetry_line_feature.py
# c. 图像检索诗句示例
python image_retrieve_peotry_demo.py
a. 可以把可工程修改为web/微信小程序
b. 如果有图像库,亦可修改为使用诗歌检索图像
c. CCPC绝句诗总共510728句,可把诗句范围缩小至诗歌经典诗句/诗篇