�
cǕi� � � � S r S rS rS r\S:X a� SSKr\" S5 V s/ s H n \" U S- 5 S - PM sn r\R
" 5 r \" S
5 H
r
\" \S S 5 M \R
" 5 \ -
r \ " S
\ S S35 \R
" 5 r \" S
5 H r
\" \S5 M \R
" 5 \ -
r \ " S\ S S35 ggs sn f )z�
Numpy-style image augmentation loop - target for rustify-ml acceleration.
Expected gain: 5-20x via Rust ndarray + rayon parallel iter.
c �r � S/[ U 5 - n[ [ U 5 5 H nX U-
U- X4' M U$ )z;Normalize a flat pixel array: (x - mean) / std per element.� ��len�range)�pixels�mean�std�result�is �Cd:\WindsurfProjects\rustify\rustify-ml\examples\image_preprocess.py�normalize_pixelsr
s= � ��U�S��[�
�F�
�3�v�;�
���Y��%��,�� � � �M� c �l � S/[ U 5 - n[ [ U 5 5 H nX U- X#' M U$ )z5Apply gamma correction to each pixel value in [0, 1].r r )r �gammar
r s r � apply_gammar s8 � ��U�S��[�
�F�
�3�v�;�
���I��&�� � � �Mr c �t � SnSn[ U[ U 5 U5 H nX0U -
nUS-
nM US:� a X4- $ S$ )z;Compute mean of a single channel from interleaved RGB data.r � � )r r )r �channels� channel_idx�total�countr s r � channel_meanr sM � � �E�
�E�
�;��F� �X�
6��
�����
��
�� 7� "�A�I�5�=�.�3�.r �__main__r Ni L � g �o@�2 g
ףp=
�?g Zd;�O�?z normalize_pixels 224x224x3: z.3fzs for 50 iterationsg������@zapply_gamma 224x224x3: )
�__doc__r
r r �__name__�timer �floatr �start�_�elapsed�print)r s 0r �<module>r% s� ��� ��/� �z���.3�M�.B�
C�.B��e�A��G�n�u�$�.B�
C�F� �I�I�K�E�
�2�Y������.� ��i�i�k�E�!�G� � (���
�5H�
I�J� �I�I�K�E�
�2�Y���F�C� � ��i�i�k�E�!�G� � #�G�C�=�0C�
D�E� �� Ds �C