https://github.com/xiaoj655/image-retrieval

基于Resnet101, 微调, 完成图像检索任务
https://github.com/xiaoj655/image-retrieval

depplearning fastapi resnet

Last synced: about 1 month ago
JSON representation

基于Resnet101, 微调, 完成图像检索任务

• Host: GitHub
• URL: https://github.com/xiaoj655/image-retrieval
• Owner: xiaoj655
• Created: 2024-12-28T12:30:53.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2025-01-02T07:03:13.000Z (over 1 year ago)
• Last Synced: 2025-03-26T07:44:26.612Z (about 1 year ago)
• Topics: depplearning, fastapi, resnet
• Language: Jupyter Notebook
• Homepage: https://movie-image-retrieval.vercel.app
• Size: 16.1 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# 微调Resnet101完成图像检索任务

通过在线网站访问试用 https://movie-image-retrieval.vercel.app

## 数据集

从223部经典电影计算平均哈希的方式, 筛选出30万张图片作为数据集

223部电影: https://www.alipan.com/s/HAxoueCChn2

约30万张原始图片: https://www.alipan.com/t/u3ovD0eXQKxqD8zkGyHD (阿里云盘, 分享不稳定)

经过预处理后的数据集, 2GB: https://msft.us.kg/movie_clip

```python

#预处理函数

transforms.Compose([

    transforms.Resize(256),

    transforms.CenterCrop(224),

])

```

## 方法

孪生网络, 对比学习的思路, 把经过图像增强后的图片和原图作为一组图像, 让相似的图像的空间嵌入接近, 不相似的远离

## 效果

![query](query.png)