Text-image search and image tagging library
Project description
ripple
(still in development)
text/Image based search and retrieval for image datasets/files
Usage guide
- Installation
pip install ripple_net
- Using ripple for text-based search
from ripple import ImageEmbedder, TextSearch # import classes
embedder = ImageEmbedder('huggan/wikiart', retrieval_type='text-image', dataset_type='huggingface') # load from a huggingface image dataset or load from a local image directory
embedded_images = embedder.create_embeddings(device="cuda", batch_size=32) # could also use 'cpu' if CUDA-enabled GPU isn't available
text_search = TextSearch(embedded_data, embedder.embed_model)
scores, images = text_search.get_similar_images(query='painting of a river', k_images=10) # specify text query for image, and number of results to return
text_search.image_grid(images, scores) #dislay grid of returned images
- For image-based retrieval .....
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
ripple_net-0.1.2.tar.gz
(148.5 kB
view details)
Built Distribution
File details
Details for the file ripple_net-0.1.2.tar.gz.
File metadata
- Download URL: ripple_net-0.1.2.tar.gz
- Upload date:
- Size: 148.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 2791621d5b5b8e41cc07c08aad2a0366262832450b482f83d7562cc08ba3cd0d |
|
| MD5 | 78b52ee46f0a33674318882c1415f082 |
|
| BLAKE2b-256 | 1213af6d925737057053264a71bc5cb6f9ae2295ec77f1a1170bd9575feca3a7 |
File details
Details for the file ripple_net-0.1.2-py3-none-any.whl.
File metadata
- Download URL: ripple_net-0.1.2-py3-none-any.whl
- Upload date:
- Size: 14.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 86a6b804b87c880e12fdeb6db68e254703394dccb9ea9bf617ed27eab9964ec8 |
|
| MD5 | beb0b1ca0c5e92c0b69a3a445b80d05e |
|
| BLAKE2b-256 | cb07358bdf3786658696aa2f098531876fab4f7415976be177b2be6896313b35 |
