You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The image dataset builder is a CLI tool developed to make it easier and faster to create image datasets to be used for deep learning. The tool achieves this by scraping images from several search engines such as duckgo, bing and deviantart. IDB also optimizes the image dataset, although this feature is optional, the user can downscale and compress the images for optimal file size and dimensions. An example dataset created using idb that contains 23.688 files weights only 559,2MBs.
10
+
The image dataset tool (IDT) is a CLI app developed to make it easier and faster to create image datasets to be used for deep learning. The tool achieves this by scraping images from several search engines such as duckgo, bing and deviantart. IDT also optimizes the image dataset, although this feature is optional, the user can downscale and compress the images for optimal file size and dimensions. An example dataset created using idt that contains 23.688 files weights only 559,2MBs.
The quickiest way to get started with IDB is running the simple "run" command. Just write in your facorite console something like:
33
+
The quickiest way to get started with IDT is running the simple "run" command. Just write in your facorite console something like:
34
34
35
35
```console
36
36
user@admin:~$ idt run -i apples
37
37
```
38
38
39
39
This will quickly download 50 images of apples. By default it uses the duckgo search engne to do so.
40
40
The run command accept the following options:
41
+
41
42
| Option | Description |
42
43
| ----------- | ----------- |
43
44
|**-i** or **--input**| the keyword to find the desired images. |
@@ -50,7 +51,7 @@ The run command accept the following options:
50
51
51
52
## Usage
52
53
53
-
IDB requires a config file that tells it how your dataset should be organized. You can create it using the following command:
54
+
IDT requires a config file that tells it how your dataset should be organized. You can create it using the following command:
54
55
55
56
```console
56
57
user@admin:~$ idt init
@@ -62,7 +63,7 @@ This command will trigger the config file creator and will ask for the desired d
62
63
Insert a name to your dataset: : My favorite cars
63
64
```
64
65
65
-
Then the tool will ask how many samples per search are required to mount your dataset. In order to build a good dataset for deep learning, many images are required and since we're using a search engine to scrape images, many searches with different keywords are required to mount a good sized dataset. This value will correspond to how many images should be downloaded at every search. In this example we need a dataset with 250 images in each class, and we'll use 5 keywords to mount each class. So if we type the number 50 here, IDB will download 50 images of every keyword provided. If we provide 5 keywords we should get the required 250 images.
66
+
Then the tool will ask how many samples per search are required to mount your dataset. In order to build a good dataset for deep learning, many images are required and since we're using a search engine to scrape images, many searches with different keywords are required to mount a good sized dataset. This value will correspond to how many images should be downloaded at every search. In this example we need a dataset with 250 images in each class, and we'll use 5 keywords to mount each class. So if we type the number 50 here, IDT will download 50 images of every keyword provided. If we provide 5 keywords we should get the required 250 images.
66
67
67
68
```console
68
69
How many samples per seach will be necessary? : 50
@@ -135,7 +136,7 @@ chevrolet impala on the road, chevrolet impala vintage car, chevrolet impala con
135
136
Then repeat the process of filling class name and its keywords until you fill all the 4 classes required.
136
137
137
138
```console
138
-
Dataset YAML file has been created sucessfully. Now run idb build to mount your dataset!
139
+
Dataset YAML file has been created sucessfully. Now run idt build to mount your dataset!
139
140
```
140
141
141
142
Your dataset configuration file has been created. Now just rust the following command and see the magic happen:
@@ -154,6 +155,9 @@ Downloading Chevrolet Impala 1967 car photos [#########################--------
154
155
155
156
At the end, all your images will be available in a folder with the dataset name. Also, a csv file with the dataset stats are also included in the dataset's root folder.
Since deep learning often requires you to split your dataset into a subset of training/validation folders, this project can also do this for you! Just run:
0 commit comments