ImagineS (Image Engine Search) tool augments an image dataset by searching a series of queries in Google Images.
Project description
ImagineS
ImagineS (Image Engine Search) tool augments an image dataset by searching a series of queries in Google Images.
Install
-
We recommend first creating an environment by using either Virtual Envs or Conda.
-
Then install the library using
pip
:pip install imagines
Usage
Example: Download apple
and water
related images by adding the query searches that would belong to each class.
from imagines import DatasetAugmentation
dataset_augmentation = DatasetAugmentation(
driver_type='chrome'
)
label_queries = {
'apple': ['apple'],
'water': ['water', 'water bottle'],
}
max_links_to_fetch = 3
sleep_between_interactions = 1
image_shape = (224, 224)
resize_images = True
output_dir = 'images/'
dataset_augmentation.augment_dataset(
label_queries,
output_dir,
max_links_to_fetch,
image_shape,
resize_images,
sleep_between_interactions
)
License
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
imagines-0.0.7.tar.gz
(5.3 kB
view hashes)