Spaces:
Running
Running
| from PIL import Image | |
| import numpy as np | |
| img_n = Image.open('scratch/cell_N.png').convert('RGBA') | |
| arr_n = np.array(img_n) | |
| # Find eye centers, lens circles/ellipses in cell_N | |
| # In cell_N: | |
| # The lenses are glasses frames. | |
| # Glasses frame color is around [204, 106, 66] or [217, 113, 74] | |
| # Sclera is around [255, 255, 255] | |
| # Let's find the centers of the two eyes in cell_N | |
| sclera_mask = (arr_n[..., 3] > 200) & (arr_n[..., 0] > 225) & (arr_n[..., 1] > 225) & (arr_n[..., 2] > 225) | |
| sy, sx = np.nonzero(sclera_mask) | |
| # Left eye in N vs Right eye in N | |
| mid_x = (sx.min() + sx.max()) / 2.0 | |
| left_sclera = sclera_mask.copy() | |
| left_sclera[:, int(mid_x):] = False | |
| right_sclera = sclera_mask.copy() | |
| right_sclera[:, :int(mid_x)] = False | |
| lsy, lsx = np.nonzero(left_sclera) | |
| rsy, rsx = np.nonzero(right_sclera) | |
| print(f"Cell N Left Eye Sclera: x=[{lsx.min()}, {lsx.max()}], y=[{lsy.min()}, {lsy.max()}], center=({lsx.mean():.1f}, {lsy.mean():.1f})") | |
| print(f"Cell N Right Eye Sclera: x=[{rsx.min()}, {rsx.max()}], y=[{rsy.min()}, {rsy.max()}], center=({rsx.mean():.1f}, {rsy.mean():.1f})") | |
| # Same for Cell C | |
| img_c = Image.open('scratch/cell_C.png').convert('RGBA') | |
| arr_c = np.array(img_c) | |
| scl_c = (arr_c[..., 3] > 200) & (arr_c[..., 0] > 225) & (arr_c[..., 1] > 225) & (arr_c[..., 2] > 225) | |
| cy, cx = np.nonzero(scl_c) | |
| mid_cx = (cx.min() + cx.max()) / 2.0 | |
| l_scl_c = scl_c.copy() | |
| l_scl_c[:, int(mid_cx):] = False | |
| r_scl_c = scl_c.copy() | |
| r_scl_c[:, :int(mid_cx)] = False | |
| clsy, clsx = np.nonzero(l_scl_c) | |
| crsy, crsx = np.nonzero(r_scl_c) | |
| print(f"Cell C Left Eye Sclera: x=[{clsx.min()}, {clsx.max()}], y=[{clsy.min()}, {clsy.max()}], center=({clsx.mean():.1f}, {clsy.mean():.1f})") | |
| print(f"Cell C Right Eye Sclera: x=[{crsx.min()}, {crsx.max()}], y=[{crsy.min()}, {crsy.max()}], center=({crsx.mean():.1f}, {crsy.mean():.1f})") | |