Datasets
Models
Downloads: 1
114 lines (110 with data), 11.4 kB
B
¢mbº= ã @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dl
m Z d dl m
Z
d dlmZmZmZmZmZmZmZ d dlmZmZ d dlmZ d dlm Z d dl m Z d d l m!Z! d
Z"d Z#d Z$e j%d
e"d e j%de"d e j%de#d e j%de"d e j%de"d e j%de#d e j%de$d de j&d< dd Z'dd Z(d1d d Z)d2d d Z*d3d!d"Z+d4d#d$Z,d5d%d&Z-d6d'd(Z.d7d)d*Z/d8d+d,Z0d-d. Z1d9d/d0Z2dS ):é N)Ú Rectangle)ÚKMeans)ÚerosionÚopeningÚclosingÚsquareÚdiskÚconvex_hull_imageÚremove_small_holes)ÚlabelÚ regionprops)Úsobel)Úcanny)Úndimage)Ú watershedé
é é Zfont)ÚsizeÚaxes)Z titlesize)Z labelsizeZxtickZytickZlegend)ZfontsizeÚfigure)é
é zfigure.figsizec C s~ g }x$t | ¡D ]}| | d | ¡ qW |jdd d |d |d d¡d |d d¡
}td|t|f ||fS ) Nú/c S s t | | d¡d | d¡
S )NÚ_é Ú.)ÚintÚfind)Ús© r ú/home/nushkash/PES1UG19CS078/4 IPCV Lab/Project - Cardiac Segmentation/cardiac-segmentation/1 - Methods with Improved Results/SegmentationFunctions.pyÚ<lambda>' ó z"readSortedSlices.<locals>.<lambda>)Úkeyr r r z,CT scan of Patient %s consists of %d slices.)ÚosÚlistdirÚappendÚsortr ÚprintÚlen)ÚpathÚslicesr ÚIDr r r! ÚreadSortedSlices" s (r. c C s t tt| S )N)ÚlistÚmapÚreadImg)r, r r r! ÚgetSliceImages, s r2 c C s8 t | ¡}t |t j¡}|r4t d¡ tj|dd |S )NzA CT Scan Image SliceÚgray)Úcmap)Úcv2ZimreadZcvtColorZCOLOR_RGB2GRAYÚpltÚtitleÚimshow)r+ Ú
showOutputÚimgr r r! r1 0 s
r1 c s t | | jd | jd ¡ dd¡}t|dd}| |¡}|j}|j}ttt |¡| t
fdd|D ¡} | | jd | jd ¡ t ¡} |rüt
jddddd d
\}
} | ¡ } | d j| d d | d d
¡ | d | ¡ | d dt| d ¡ | fS )zº
Apply KMeans on an image with the number of clusters K
Input: Image, Number of clusters K
Output: Dictionary of cluster center labels and points, Output segmented image
r r éÿÿÿÿ)Z
n_clustersÚverbosec s g | ] } | qS r r )Ú.0r )Ú
center_labelsr r! ú
<listcomp>K s z imgKMeans.<locals>.<listcomp>é T)r r )ÚsharexÚshareyÚfigsizer3 )r4 zOriginal ImagezImage after KMeans (K = ú))ÚnpZreshapeÚshaper ZfitZlabels_Zcluster_centers_ÚdictÚzipZarangeÚarrayZastyper r6 ÚsubplotsÚravelr8 Ú set_titleÚstr) r: ÚKr9 Z
showHistogramZimgflatZkmeansZkmmodelÚlabelsZcentersÚoutputÚfigr r )r> r! Ú imgKMeans: s $
rR c
C s¾ t | ddd\}}tdd | ¡ D }|d |d d }t | |d tj¡\}}|r¶tjd
dd d d d
\}} | ¡ } | d j | dd | d
d¡ | d
j |dd | d
d¡ ||fS )z~
Preprocess the image by applying truncated thresholding using KMeans
Input: Image
Output: Preprocessed image
é r )r9 c S s g | ] }|d qS )r r )r= Úir r r! r? c s z#preprocessImage.<locals>.<listcomp>r; éþÿÿÿr@ éÿ r T)r r )rA rB rC r3 )r4 zOriginal Imagez$Processed Image - After Thresholding) rR ÚsortedÚvaluesr5 Ú thresholdZ
THRESH_TOZEROr6 rJ rK r8 rL )
r: r9 Ú centroidsZ
segmented_imgZsorted_center_valuesrY ÚretvalÚprocImgrQ r r r r! ÚpreprocessImage[ s r] c C s. t | |dt j¡\}}t|td}t|td}t|}t |¡}|r*t j
dddddd\} }
|
¡ }
|
d d ¡ |
d j
| d
d |
d d
¡ |
d j
|d
d |
d d¡ |
d j
|d
d |
d d¡ |
d j
|d
d |
d d¡ |
d j
|d
d |
d d¡ |
d j
|d
d |S )NrV rS é r@ T)é r )rA rB rC r zOriginal Imager3 )r4 r z On Performing Thresholding'sz On Opening with Square SE (3)zOn Opening with Disk SE (4)z'Outer Boundary Delineation with Canny'sr zForeground Mask)r5 rY Ú
THRESH_BINARYr r r r ÚndiÚbinary_fill_holesr6 rJ rK rL r8 ) r: Ú fg_thresholdr9 r[ Z init_fg_maskZfg_mask_openedZfg_mask_opened2ZedgesÚfg_maskrQ r r r r! ÚgetForegroundMaskv s*
re c C s´ | ¡ }xJt| jd D ]8}x2t| jd D ] }|| | dkr,d|| |< q,W qW t ||dtj¡\}}t|td} t| t d}
t
|
¡} t | } t
| }
g }x"|
D ]}|jdk rª| |j ¡ qªW tj| jtjd}xJt| jd D ]8}x2t| jd D ] }| | | |krþd|| |< qþW qêW | t |¡ }tt|dkrPt|}n| ¡ }|t |¡ t |¡ }t|td}t|t d }t |}t
|}i }x|D ]}|j||j < q¦W |rðt||jd
}t ||kt d¡t d¡¡}ntj| jtjd}|r¨tjd d d d d
d\} } | ¡ } | d d¡ | d j | dd | d d¡ | d j |dd | d d¡ | d j |dd | d d¡ | d j | dd | d d¡ | d j |
dd | d d¡ | d j | dd | d d¡ | d j |dd | d d¡ | d j |dd | d d¡ | d j |dd | d d ¡ | d j |dd | d d ¡ | d j |dd | d d!¡ | d j |dd ||||fS )"Nr r rV r@ é iÜ )Údtyper r^ )r$ rS T)é rh )rA rB rC zOriginal Imager3 )r4 zEnhanced ImagezInitial Lung Maskz On Opening with Square SE (2)zOn Closing with Disk SE (6)zOn Filling Regionsz Trachea Mask/Primary Bronchié z Lung Maské zConvex Hull of Lung Maské z Initial Intermediate Heart Maskr z On Opening with Square SE (3)é zIntermediate heart mask) ÚcopyÚrangerF r5 rY ZTHRESH_BINARY_INVr r r r ra rb r r Úarear' rE ÚzerosÚuint8ÚinvertÚsumr ÚmaxÚgetÚwherer6 rJ rK rL r8 ) r: rd rc r9 ZenhancedrT Újr[ Zinitial_lung_maskZ lung_mask_opZlung_mask_opclZlung_mask_opclrmZ label_imgZ lung_regionsZtrachea_labelsÚ trachea_maskZrowZcolÚ lung_maskÚ ch_lung_maskZinitial_int_heart_maskZint_heart_mask_op1Zint_heart_mask_op2Zheart_label_imgÚ
heart_regionsÚareasZ heart_labelÚint_heart_maskrQ r r r r! ÚgetLungTracheaMasks s
r~ c " C s6 | ¡ s||fS | ¡ }xJt| jd D ]8}x2t| jd D ] }|| | dkr<d|| |< q<W q(W t|ddd\}}t| ¡ d }t ||dtj ¡\} }
t
|
t d} t | } t
| }
g }i }i }x:|
D ]2}| |j ¡ |j||j < |j|j|jg||j < qÌW t||jd}| |¡ t | |kt d¡t d¡¡}t |}t
|}|d j}|| }|| | }|dk rtj| jtjd }d}nòd}|| d \}}|| d } |t | ¡d
|| d } |t | ¡d
|| d } | ¡ } x~tt | ¡| jd D ]d}|t |¡kr@xLt| jd D ]}d| | |< q(W n&x$tt |¡D ]}d| | |< qPW qW |t | ¡ }|r.t j ddd d d
d\} }!|! ¡ }!|!d !d¡ |!d j"|dd |!d !d¡ |!d j"|dd |!d !d¡ |!d "|¡ |!d !d¡ |!d j"|
dd |!d !d¡ |!d j"|dd |!d !d¡ |!d j"|dd |!d !d¡ |!d j"|dd |rî|!d j#|| f|| fdd d |!d j#||d dd |!d j#| | d dd |!d! !d"¡ |!d! j"| dd n |!d! !d"¡ |!d! j"| dd |!d# !d$¡ |!d# j"|dd ||fS )%Nr r rS )r9 rV rh )r$ g à?)rg g333333ã?r@ T)r r )rA rB rC zIntermediate Heart Maskr3 )r4 zIntermediate Heart Segmentz-Intermediate Heart Segment on K-Means (K = 3)z
Spine Maskr^ z On Closing with Disk SE (20)r z On Opening with Square SE (4)rf z Centroid and uppermost pointz-rg ø?)Z linewidthz.g)Z
markersizez.bri zConvex Hull of Spine Maskrj z
Heart Mask)$Úanyrm rn rF rR rt rX r5 rY r` r r r r r' ro ZcentroidÚ orientationZaxis_major_lengthru ÚremoverE rv rq rp ÚmathZsinZcosZceilrr r6 rJ rK rL r8 Zplot)"r: r} r9 Zint_heart_pixelrT rw rZ Zsegmented_heart_imgZspine_thresholdr[ Zinitial_spine_maskZ bone_maskZ label_spineZ
spine_regionsrO r| Zgeometric_measuresZ spine_labelÚ
spine_maskZ label_heartr{ Zheart_region_areaZspine_region_areaZ
frac_heartÚ
heart_maskZmake_spine_maskZy0Zx0r Zx2Zy2Zchull_spine_maskrQ r r r r! ÚchullSpineMask s
r
c C s< t j| | |d}|r8tjdd t d¡ tj|dd |S )N)Úmask)r r )rC zSegmented Heartr3 )r4 )r5 Ú bitwise_andr6 r r7 r8 )r: r r9 Ú seg_heartr r r! Ú segmentHeart s
r c C s¤ t j| | |d}t j| | |d}t j| | |d}|rtjdddddd\}\} }
} | d¡ | j|d d
|
d ¡ |
j|d d
| d ¡ | j|d d
|||fS )
N)r r rS )é rf TF)rC rA rB zSegmented Heartr3 )r4 zSegmented LungszSegmented Trachea)r5 r r6 rJ rL r8 ) r: r ry rx r9 r Z seg_lungsZ seg_trachearQ Zax1Zax2Zax3r r r! ÚsegmentHeartLungsTrachea s
r c sP t j fdd|D dd}t |¡}t |¡}|| dkr@|S ||| }|S )Nc s g | ]} d t jf | qS ).)rE Znewaxis)r= Zcolor)r r r! r? ¤ s z"applyMaskColor.<locals>.<listcomp>r@ )Zaxisr )rE Z concatenateZamaxZamin)r Z
mask_colorZmaskedZmaxValueZminValuer )r r! ÚapplyMaskColor¢ s
r c C sú t dddg¡}t dddg¡}t dddg¡}t||}t||} t||}
|| |
} |rîtjdddd\} }
|
¡ }|d d¡ |d j| dd |d d
¡ |d |¡ |d d ¡ |d | ¡ |d d
¡ |d | ¡ || |
| fS )Né r r@ )r r )rC zOriginal Imager3 )r4 r z
Heart Maskz Lung MaskrS ZMasks)rE rI r r6 rJ rK rL r8 )r: r ry rx r9 Zheart_mask_colorZlung_mask_colorZtrachea_mask_colorÚ
heart_coloredÚ lung_coloredÚtrachea_coloredÚ
colored_masksrQ r Zaxr r r! ÚgetColoredMasks² s&
r )r )r r )r )r )r )r )r )r )r )3r% r5 r ZnumpyrE ZpandasZpdZmatplotlib.pyplotZpyplotr6 Zmatplotlib.patchesr Zsklearn.clusterr Zskimage.morphologyr r r r r r r
Zskimage.measurer r Zskimage.filtersr
Zskimage.featurer Zscipyr ra Zskimage.segmentationr Z
SMALL_FONTZ MEDIUM_FONTZ
LARGE_FONTZrcZrcParamsr. r2 r1 rR r] re r~ r
r r r r r r r r! Ú<module> sH $
!
%
l
|