Skip to content

An Image Search and Pair system base on PaddlePaddle.

License

Notifications You must be signed in to change notification settings

AgentMaker/Paddle-IMSP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Paddle-IMSP

GitHub forks GitHub Repo stars GitHub release (latest by date including pre-releases) GitHub
An Image Search and Pair system base on PaddlePaddle.

Search Web Image by text and pair similar images by image.

Base project:【Paddle-CLIP】.

Project Information

  • This project is limited to non-commercial scenarios

  • All the retrieved and matched web images are from the website: 【unsplash】

  • If you need to use the original image, please move to 【unsplash】 Download

  • Thanks to 【unsplash】 open source large-scale image dataset

  • Thanks to 【openai/CLIP】 open source model code and pretrained model parameters

Install Package

  • Install by pip:
$ pip install paddleimsp

Quick Start

from imsp import IMSP

# Load the engine
# The first load will automatically download the pretrained model and images database
imsp_engine = IMSP()
# Search Web Image by text
# You can use Chinese or English
photo_urls = imsp_engine.im_search('sky')

Preview photos:

Original image:Unsplash Link

Original image:Unsplash Link

Original image:Unsplash Link

# Pair similar images by image
photo_urls = imsp_engine.im_pair('fruit.jpg')

Preview photos:

Original image:Unsplash Link

Original image:Unsplash Link

Original image:Unsplash Link

# Use the images and texts to search and pair images
photo_urls = imsp_engine.im_search_pair(images=['road.jpg'], texts=['cars', 'peoples'], topk=5)

Preview photos:

Original image:Unsplash Link

Original image:Unsplash Link

Original image:Unsplash Link

Contact us

Email : [email protected]
QQ Group : 1005109853