
Image Retrieval系统:利用Pytorch与Flask构建,致谢image_retrieval_platform
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目为一个基于Pytorch和Flask开发的图像检索系统。旨在实现高效、准确的图片搜索功能,并感谢image_retrieval_platform的支持。
图像检索数据集:CIFAR10;模型包括ResNet50、以及由一个包含一万个网页爬取的无标签图片组成的网络,使用简单的CNN(在CIFAR10上训练)或ResNet50(在ImageNet上预训练)。请先安装requirements.txt文件中的依赖。
例子1:如果您需要预训练的模型,则可以获取并将其放置于路径./retrieval/models/。运行以下命令:
```
python image_retrieval.py
# 第一次运行时,应添加--update来生成大图片数据集的缩略图和特征。
```
例子2:我们在CIFAR-10数据集上实现了一个版本。
全部评论 (0)
还没有任何评论哟~


