37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from icrawler.builtin import BingImageCrawler
|
|
from icrawler.builtin import GoogleImageCrawler
|
|
|
|
# we are building cats image detection that's why we put cat here
|
|
# if you want some other images then put that name in classes list
|
|
classes = [
|
|
"horse standing in white background",
|
|
"horse running in white background",
|
|
"horse sleeping in white background",
|
|
"horse eating in white background",
|
|
"horse shitting in white background",
|
|
"horse sitting in white background",
|
|
"horse racing in white background",
|
|
"horse head in white background",
|
|
"horse body in white background",
|
|
"horse legs in white background",
|
|
"horse ears in white background",
|
|
"horse eyes in white background",
|
|
"horse tail in white background",
|
|
"any horse in white background",
|
|
]
|
|
number = 99999
|
|
# here root directory is find your root directory there u will find
|
|
# new file name data in which all images are saved.
|
|
for c in classes:
|
|
# bing_crawler = BingImageCrawler(storage={"root_dir": f'p/{c.replace(" ",".")}'})
|
|
# bing_crawler.crawl(keyword=c, filters=None, max_num=number, offset=0)
|
|
|
|
google_crawler = GoogleImageCrawler(storage={"root_dir": f'p/{c.replace(" ",".")}'})
|
|
google_crawler.crawl(
|
|
keyword=c,
|
|
offset=0,
|
|
max_num=99999,
|
|
min_size=(50, 50),
|
|
max_size=None,
|
|
)
|