image_names = os.listdir('data/train_masks/')
np.random.shuffle(image_names)
for image_name in image_names[:10]:
mask = Image.open('data/train_masks/{}'.format(image_name))
mask = np.asarray(mask)
mask_rotated = rotate(mask, angle=15)
mask_reverse_rotated = rotate(mask_rotated, -15)
fig = plt.figure(figsize=(18, 5))
a = fig.add_subplot(1, 3, 1)
plt.imshow(mask)
a = fig.add_subplot(1, 3, 2)
plt.imshow(mask_rotated)
a = fig.add_subplot(1, 3, 3)
plt.imshow(mask_reverse_rotated)
print ('dice: ', dice_coeff(mask, mask_reverse_rotated))
