Skip to content

Instantly share code, notes, and snippets.

@InnovArul
Created January 1, 2020 11:22
Show Gist options
  • Save InnovArul/9923bb2b19f98f251998d8d74d3c171b to your computer and use it in GitHub Desktop.
Save InnovArul/9923bb2b19f98f251998d8d74d3c171b to your computer and use it in GitHub Desktop.
import imgaug as ia
import imgaug.augmenters as iaa
print("loaded")
prob = 0.5
sometimes = lambda aug: iaa.Sometimes(prob, aug)
output_shape=(473,473)
seq = iaa.Sequential([
# apply the following augmenters to most images
iaa.Fliplr(0.5),
sometimes(iaa.Affine(
scale={"x": (0.8, 1.2), "y": (0.8, 1.2)},
translate_percent={"x": (-0.2, 0.2), "y": (-0.2, 0.2)},
order=[0, 1],
cval=(0, 255),
mode=ia.ALL
)),
iaa.Resize({"height": output_shape[0], "width": "keep-aspect-ratio"}),
iaa.CropToFixedSize(width=output_shape[1], height=output_shape[0], position="center"),
iaa.PadToFixedSize(width=output_shape[1], height=output_shape[0], pad_mode=ia.ALL),
# incase if the resizing is missed, resize the images explicity
iaa.Resize({"height": output_shape[0], "width": output_shape[1]})
])
# creating dummy images
import numpy as np
for i in range(100):
img = np.random.rand(709,450,3)
seg = np.ones((1,709,450,1))
print("augmenting", i)
iaug, saug = seq(image=img, segmentation_maps=seg.astype(np.int32))
@InnovArul
Copy link
Author

augmenting 1
augmenting 2
augmenting 3
augmenting 4
augmenting 5
augmenting 6
augmenting 7
augmenting 8
augmenting 9

GDB backtrace:
Thread 26 "python" received signal SIGSEGV, Segmentation fault.
[Switching to Thread 0x7fff737fe700 (LWP 34404)]
__GI___libc_free (mem=0x4015) at malloc.c:2951
2951 malloc.c: No such file or directory.
(gdb) backtrace
#0 __GI___libc_free (mem=0x4015) at malloc.c:2951
#1 0x00007fffe6d29a39 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#2 0x00007fffe69a0ef5 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#3 0x00007fffe70a16c5 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#4 0x00007fffe6de97e9 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#5 0x00007fffe70a6c31 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#6 0x00007fffe70aa5a7 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#7 0x00007fffe6de963e in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#8 0x00007fffe6dea04a in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#9 0x00007fffe6dead1d in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#10 0x00007fffe6deae89 in ?? () from /home/arul/miniconda3/envs/vos/lib/python3.7/site-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so
#11 0x00007ffff7bc16ba in start_thread (arg=0x7fff737fe700) at pthread_create.c:333
#12 0x00007ffff78f741d in clone () at ../sysdeps/unix/sysv/linux/x86_64/clone.S:109

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment