PhysicalBackdoorDetectors/transformations/shadow_ellipse.py at master · AdelaideAuto-IDLab/PhysicalBackdoorDetectors · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import cv2
import numpy as np

import transformations.shadow_mask as mask


def add_n_ellipses_light(image, intensity = 0.5, blur_width = 6, n = 1, WHITE = 255):
	inverted_colors = WHITE - image
	inverted_shadow = add_n_ellipses_shadow(inverted_colors, intensity, blur_width, n)

	return WHITE - inverted_shadow


def add_n_ellipses_shadow(image, intensity = 0.5, blur_width = 6, n = 1):
	for i in range(n):
		image = add_ellipse_shadow(image,
					   intensity = intensity,
					   blur_width = blur_width,
					   )

	return image


def add_ellipse_light(image, intensity = 0.5, blur_width = 6, WHITE = 255):
	inverted_colors = WHITE - image
	inverted_shadow = add_ellipse_shadow(inverted_colors, intensity, blur_width)

	return WHITE - inverted_shadow


def add_ellipse_shadow(image, intensity = 0.5, blur_width = 6, WHITE = 255):

	shadow_mask = np.zeros(image.shape[: 2], dtype=np.uint8)
	shadow_mask.fill(WHITE)
	ellipse = __get_multiple_ellipses(shadow_mask)

	return mask.apply_shadow_mask(image, blur_width, intensity, ellipse)


def __get_multiple_ellipses(image):

	LIGHT_BLACK     = 50
	DARK_GRAY       = 100
	GRAY            = 150
	LIGHT_GRAY      = 200
	DARK_WHITE      = 250

	h, w = image.shape[ : 2]
	center = int(w * np.random.uniform()), int(h * np.random.uniform())
	random_h = np.random.uniform() * h
	random_w = np.random.uniform() * w
	axes1 = int(random_h * 0.2), int(random_w * 0.2)
	axes2 = int(random_h * 0.4), int(random_w * 0.4)
	axes3 = int(random_h * 0.6), int(random_w * 0.6)
	axes4 = int(random_h * 0.8), int(random_w * 0.8)
	axes5 = int(random_h), int(random_w)
	angle = 360 * np.random.uniform()

	ellipse = get_single_ellipse(image,   center, axes5, angle, DARK_WHITE)
	ellipse = get_single_ellipse(ellipse, center, axes4, angle, LIGHT_GRAY)
	ellipse = get_single_ellipse(ellipse, center, axes3, angle, GRAY)
	ellipse = get_single_ellipse(ellipse, center, axes2, angle, DARK_GRAY)

	return get_single_ellipse(ellipse, center, axes1, angle, LIGHT_BLACK)


def get_single_ellipse(image, center, axes, angle, color):
	start_angle = 0
	end_angle = 360
	thickness = -1

	return cv2.ellipse(image, center, axes, angle, start_angle, end_angle,
			   color, thickness)