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.1.tar.gz
(148.5 kB
view details)
Built Distribution
File details
Details for the file ripple_net-0.1.1.tar.gz.
File metadata
- Download URL: ripple_net-0.1.1.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 | 4494c2397634e371bc2a1d0a3ec6177215da88ac514d0fb62f25ff6fa3b9b903 |
|
| MD5 | f3f6842dd8a5854052357d875c703967 |
|
| BLAKE2b-256 | 1d03b3a65759f380efedc98aaefa172a23122542bf0be0984017a7aaa5f01326 |
File details
Details for the file ripple_net-0.1.1-py3-none-any.whl.
File metadata
- Download URL: ripple_net-0.1.1-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 | 82d12498c488c9dde41f2e367c104ce730b33a6881a1867e4da6154e3b2410b4 |
|
| MD5 | 9e0a14f09a0b23a4cbf6f4ca28654562 |
|
| BLAKE2b-256 | 761492edf3833bc37aba44a858498c051bd9b9c444e3275efdad5997c2153206 |
