Datasets
Models
Downloads: 1
26 lines (25 with data), 3.2 kB
3 ï `v ã @ sh d dl Zd dlmZ d dlmZ d dlmZmZ d dlm Z G dd dZ dd Z d d Z dd dZ dS )é N)Úrotate)Úgaussian_filter)Úinterp1dÚinterp2d)Úcenter_of_massc @ s& e Zd Zdd Zd ddZdd Zd S ) ÚMyocardialStrainc C s^ || _ || _t|jdks tt|jdks0t|j|jd d ksHt|jd dksZtd S )Né é é éÿÿÿÿ)ÚmaskÚflowÚlenÚshapeÚAssertionError)Úselfr r © r úP/tf/Dropbox (Partners HealthCare)/ubuntu/docker/repos/DeepStrain/utils/strain.pyÚ__init__ s zMyocardialStrain.__init__r r c C s t | j|k\}}}| jj\}} } t| j||| _t| j||| _tj| jdd\} } } tj tj | |||\}}}tj tj | |||\}}}tj tj | |||\}}}tj || | ddf| _ xòt |D ]æ}xàt | D ]Ô}xÎt | D ]Â}||||f ||||f ||||f g||||f ||||f ||||f g||||f ||||f ||||f gg}tj|tjd }dtj|j|tjd } | j |||f | 7 < qèW qÚW qÌW | j d d d d d d ddf | _| j| j d d d d d d d d d d f \| _| _d S )Nr r g à?é r )r r r Ú_roll_to_centerr Zflow_rotZmask_rotÚnpÚ array_splitÚgradientÚsqueezeÚzerosZE_cartÚrangeÚarrayÚidentityÚmatmulÚTZEzzÚ_convert_to_polarÚErrÚEcc) r ÚdxÚdyÚdzZlv_labelÚcxÚcyÚczÚnxÚnyÚnzÚuxÚuyÚuzZUxxZUxyZUxzZUyxZUyyZUyzZUzxZUzyZUzzÚiÚjÚkZUgradÚFÚer r r Úcalculate_strain s( &&* $"z!MyocardialStrain.calculate_strainc C sn t |jd d d }tj| jj}tj| jj}x0t| jjd D ]}tjtj|}tjtj|}|d d d d |ddf |d d d d |ddf |d d d d |ddf |d d d d |ddf f\}} } } |d d d d |f ||| || ||| || 7 < |d d d d |f | | | || || | || 7 < qFW ||fS )Nr r r r ) Ú _polar_gridr r r r r ÚcosÚdeg2radÚsin) r ÚEÚphir" r# r2 r7 r9 ZExxZExyZEyxZEyyr r r r! 1 s lBLz"MyocardialStrain._convert_to_polarN)r r r r )Ú__name__Ú __module__Ú __qualname__r r5 r! r r r r r s r c C s t j| |dd} t j| |ddS )Nr )Úaxisr )r Úroll)ÚxÚrxÚryr r r Ú_rollB s rD c C s6 | j d d \}}t| t|d | t|d | S )Nr )r rD Úint)rA r' r( r* r+ r r r r F s r é c C sr t jt j| d | d | t j| d |d |\}}t jt j||d j}t j|d |d d }||fS )Nr é´ g:0âyE>)r ÚmeshgridÚlinspaceÚrad2degÚarctan2r Úsqrt)r* r+ rA Úyr; Úrr r r r6 J s 8r6 )rF rF )Únumpyr Ú scipy.ndimager r Úscipy.interpolater r Zscipy.ndimage.measurementsr r rD r r6 r r r r Ú<module> s ;