Team Ai
Modelpublic

fernandoperlar/preprocessing_image

sourceHugging Faceupdated 5y agoView on Hugging Face
0likes
rootsrcscripts__pycache__Data.cpython-38.pyc
Data.cpython-38.pyc33 linesDownload Raw Back to __pycache__
1U

2�T�aV
�@s^ddlZddlZddlZddlZddlmZ	ddl3mZddlm
Z
ddlTGdd�d�ZdS)�N)�model_selection)�
preprocessing�)�*c@sPeZdZdd�Zddd�Zdd	�Zdd4d�Zdd
�Zdd�Zdd�Z	dd�Z5dS)�DatacCsD|�|�|_|jj��\|j_|_|jj�t�|j_d\|_|_dS)N)NN)	�_Data__extract_images�images�categoryZ	factorize�labels�astype�str�training�test)�self�path�r�<c:\Users\ferna\Desktop\mammo-echo-thermo\src\scripts\Data.py�__init__sz
Data.__init__�333333�?TFcCs"tj|j|d||r|jjndd�S)N�*)�	test_sizeZrandom_state�shuffle�stratify)r�train_test_splitrr	)rrrrrrrrs�zData.train_test_splitc6Cs8|j��j}t|�d|�dt�|t|�d����dS)Nz: � �)r	Zvalue_counts�values�print�np�round�len)r�data�name�amountrrr�count_labelsszData.count_labelscCs�tjjddd�}tjjdd�}ddddd	d7�}|jf||jd|dd
���}|jf||jd|dd
���}|jf||jddd���}|||fS)Ngp?g�������?)�rescaleZvalidation_split)r%�imager	)��i�rgbZcategorical)Zx_colZy_colZtarget_sizeZ8color_modeZ9class_mode�10r
)�	dataframe�11batch_sizerZsubset�12validationrF)r*r+r)rr&ZImageDataGeneratorZflow_from_dataframer
r)rrZ
train_datagenZtest_datagenZgenerator_propertiesZtrain_generatorZvalidation_generatorZtest_generatorrrr�image_generators:��13	�14	�15zData.image_generatorcCsbt�|�r|����}n|��}|��}t�|tj�}t�|||�}|��}d|t�	|dk�<|S)Nr)16�tfZ	is_tensor�numpy�copy�cvZcvtColorZ
COLOR_RGB2HSVZinRanger�where)rr&�lower�upperZ17temp_imageZ	hsv_image�mask�resultrrr�detectColorGs18zData.detectColorc	Cs:g}|D]"}|�tj|�|||�dd��qtj|dd�S)Nr��axis)�appendr�expand_dimsr7�concatenate)rrr3r4�results�imgrrr�getImageTensorVs zData.getImageTensorc
	s|��|��\}}tjdddd�\�}��|�|D]}|��q6|d����}�fdd�|D�}	t|	�D]�\}19}|�t	|j20tj||21dd	���
��|jdd22d�}t|�D]N\}}|�t|���||�|23�|�t|�|�
��|�d�|��q�qndS)
N�rT)�nrows�ncolsZconstrained_layoutrcsg|]}��|��qSr)Z
add_subfigure)�.0�gs��figrr�24<listcomp>jsz$Data.show_images.<locals>.<listcomp>�����r8�)rArB�off)�reset�next�pltZsubplotsZsuptitle�removeZget_subplotspecZget_gridspec�	enumeraterr25r�argmax�titleZimshow�listr�	set_titler9Zplot)
r�	generator�filtersr"r>�labelZaxs�axZgridspecZsubfigs�rowZsubfig�colrrEr�show_images^s 2627&28zData.show_imagescCsTg}t�|�D]2}t�||�D]}|�||d||g�q qtj|ddgd�S)N�/r&r	)�columns)�os�listdirr:�pd�	DataFrame)rrrr	�filenamerrrZ__extract_imagesxs29zData.__extract_imagesN)rTF)T)�__name__�30__module__�__qualname__rrr$r-r7r?rZrrrrrr31s32	33(r)r]�pandasr_r/rZcv2r1�matplotlib.pyplot�pyplotrMZsklearnrZkerasr�Miscrrrrr�<module>s