Text-image search and image tagging library
Project description
ripple
(still in development)
Search image datasets with words, or even label images automatically
Install with pip install ripple_net
(I name things weirdly :/)
Usage guide
from ripple import ImageEmbedder, ImageSearch # import classes
embedder = ImageEmbedder('huggan/wikiart', is_hfdataset=True) # 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 = ImageSearch(embedded_data, data_indexer.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
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.0.tar.gz
(9.3 kB
view details)
Built Distribution
File details
Details for the file ripple_net-0.1.0.tar.gz.
File metadata
- Download URL: ripple_net-0.1.0.tar.gz
- Upload date:
- Size: 9.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 90933a569d2016f2cfb6695aaa46069eb63c39835a28adb2a4bbd54c80f2e5f9 |
|
| MD5 | ca6fcaae4c097e277ae8a4c5e6f3ec34 |
|
| BLAKE2b-256 | e2d8a9c489b1afb439e9fdd0aa4c240d0277c25fcc538e203d8483100ec1c166 |
File details
Details for the file ripple_net-0.1.0-py3-none-any.whl.
File metadata
- Download URL: ripple_net-0.1.0-py3-none-any.whl
- Upload date:
- Size: 11.7 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 | 2997ef71fd735017dc15c92ebceaa5fe5a4c16b7bd45d999cd185d6b0010de4c |
|
| MD5 | 85d7e76584c3d4de4858976aee1293e2 |
|
| BLAKE2b-256 | 75c29a1cdea3baa2150ca90997a2755e40222638d339dadc70233a9462383707 |
