aleju--imgaug
75 行
3.5 KiB
Python
75 行
3.5 KiB
Python
import imgaug as ia
|
|
import imgaug.augmenters as iaa
|
|
import imageio
|
|
import cv2
|
|
import numpy as np
|
|
|
|
|
|
def main():
|
|
urls_small = [
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/0/08/"
|
|
"Physicist_Stephen_Hawking_in_Zero_Gravity_NASA.jpg/"
|
|
"320px-Physicist_Stephen_Hawking_in_Zero_Gravity_NASA.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/5/5d/"
|
|
"Barack_Obama_family_portrait_2011.jpg/320px-Barack_Obama_"
|
|
"family_portrait_2011.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/f/f6/"
|
|
"Pahalgam_Valley.jpg/320px-Pahalgam_Valley.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/a/a4/"
|
|
"Iglesia_de_Nuestra_Se%C3%B1ora_de_La_Blanca%2C_Cardej%C3%B3n%2C_"
|
|
"Espa%C3%B1a%2C_2012-09-01%2C_DD_02.JPG/320px-Iglesia_de_Nuestra_"
|
|
"Se%C3%B1ora_de_La_Blanca%2C_Cardej%C3%B3n%2C_Espa%C3%B1a%2C_"
|
|
"2012-09-01%2C_DD_02.JPG",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/9/94/"
|
|
"Salad_platter.jpg/320px-Salad_platter.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/1/1c/"
|
|
"Squirrel_posing.jpg/287px-Squirrel_posing.jpg"
|
|
]
|
|
urls_medium = [
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/0/08/"
|
|
"Physicist_Stephen_Hawking_in_Zero_Gravity_NASA.jpg/"
|
|
"640px-Physicist_Stephen_Hawking_in_Zero_Gravity_NASA.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/5/5d/"
|
|
"Barack_Obama_family_portrait_2011.jpg/640px-Barack_Obama_"
|
|
"family_portrait_2011.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/f/f6/"
|
|
"Pahalgam_Valley.jpg/640px-Pahalgam_Valley.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/a/a4/"
|
|
"Iglesia_de_Nuestra_Se%C3%B1ora_de_La_Blanca%2C_Cardej%C3%B3n%2C_"
|
|
"Espa%C3%B1a%2C_2012-09-01%2C_DD_02.JPG/640px-Iglesia_de_Nuestra_"
|
|
"Se%C3%B1ora_de_La_Blanca%2C_Cardej%C3%B3n%2C_Espa%C3%B1a%2C_"
|
|
"2012-09-01%2C_DD_02.JPG",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/9/94/"
|
|
"Salad_platter.jpg/640px-Salad_platter.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/1/1c/"
|
|
"Squirrel_posing.jpg/574px-Squirrel_posing.jpg"
|
|
]
|
|
urls_large = [
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/0/08/"
|
|
"Physicist_Stephen_Hawking_in_Zero_Gravity_NASA.jpg/"
|
|
"1024px-Physicist_Stephen_Hawking_in_Zero_Gravity_NASA.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/5/5d/"
|
|
"Barack_Obama_family_portrait_2011.jpg/1024px-Barack_Obama_"
|
|
"family_portrait_2011.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/f/f6/"
|
|
"Pahalgam_Valley.jpg/1024px-Pahalgam_Valley.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/a/a4/"
|
|
"Iglesia_de_Nuestra_Se%C3%B1ora_de_La_Blanca%2C_Cardej%C3%B3n%2C_"
|
|
"Espa%C3%B1a%2C_2012-09-01%2C_DD_02.JPG/1024px-Iglesia_de_"
|
|
"Nuestra_Se%C3%B1ora_de_La_Blanca%2C_Cardej%C3%B3n%2C_"
|
|
"Espa%C3%B1a%2C_2012-09-01%2C_DD_02.JPG",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/9/94/"
|
|
"Salad_platter.jpg/1024px-Salad_platter.jpg",
|
|
"https://upload.wikimedia.org/wikipedia/commons/thumb/1/1c/"
|
|
"Squirrel_posing.jpg/574px-Squirrel_posing.jpg"
|
|
]
|
|
|
|
image = imageio.imread(urls_medium[1])
|
|
|
|
augs = [image] + iaa.Cartoon()(images=[image] * 15)
|
|
ia.imshow(ia.draw_grid(augs, 4, 4))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|