Center Deviation Measurement of Color Contact Lenses Based on a Deep Learning Model and Hough Circle Transform

. 2023 Jul 19;23(14):6533. doi: 10.3390/s23146533

Algorithm 1: AugmentTrainSet(img, iteration, augmented_images)

//Augment train images from an original image

Input

(1)

i m g

: Original image in RGB channel

(2)

iteration

: The number of iterations to perform

Output

(3)

augmented_images

: List of augmented images

Begin

augmented_images \leftarrow empty list

n \leftarrow height of img

m \leftarrow width of img

For i = 1 to iteration do

{i m g}_{c o p i e d} \leftarrow c o p y o f i m g

{r a n d}_{f l i p} \leftarrow r a n d o m i n t e g e r i n t h e r a n g e - 1 t o 1

{r a n d}_{r e d}, {r a n d}_{g r e e n}, {r a n d}_{b l u e} \leftarrow r a n d o m i n t e g e r i n t h e r a n g e - 10 t o 10

I f {r a n d}_{f l i p} i s - 1 t h e n

{i m g}_{c o p i e d} \leftarrow i m g [n - i, m - j] f o r a l l i i n 1 t o n a n d j i n 1 t o m

10.

E l s e i f {r a n d}_{f l i p} i s 0 t h e n

11.

{i m g}_{c o p i e d} \leftarrow i m g [i, m - j] f o r a l l i i n 1 t o n a n d j i n 1 t o m

12.

E l s e i f {r a n d}_{f l i p} i s 1 t h e n

13.

{i m g}_{c o p i e d} \leftarrow i m g [n - i, j] f o r a l l i i n 1 t o n a n d j i n 1 t o m

14.

F o r e a c h p i x e l p i n r e d c h a n n e l o f {i m g}_{c o p i e d}

15.

p \leftarrow c l i p (p + {r a n d}_{r e d}, 0, 255)

16.

E n d f o r

17.

F o r e a c h p i x e l p i n g r e e n c h a n n e l o f {i m g}_{c o p i e d}

18.

p \leftarrow c l i p (p + {r a n d}_{g r e e n}, 0, 255)

19.

E n d f o r

20.

F o r e a c h p i x e l p i n b l u e c h a n n e l o f {i m g}_{c o p i e d}

21.

p \leftarrow c l i p (p + {r a n d}_{b l u e}, 0, 255)

22.

E n d f o r

23.

a u g m e n t e d_i m a g e s . a p p e n d ({i m g}_{c o p i e d})

24.

E n d f o r

25.

R e t u r n a u g m e n t e d_i m a g e s

END