Downloads: 1
25 lines (25 with data), 5.4 kB
ó
¸Yc @ sÙ d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d d g d d d d Z d Z
d Z d Z d
Z
d d d g d Z d d d Z d d
Z d Z d Z d Z d S( iÿÿÿÿNi gÍÌÌÌÌÌì?gñ?i c C s< g } t j j | d | d t j d } t j j | d | d t j d } t j j | d | d t j d } t j j | | } x t t | D] }
t | |
} |
d k rá d g t | } n d g t | } t | | | }
t |
| | | g | } | j t
| q© W| S( Ni g f@i i i ( t npt randomt uniformt pit ranget lent convertBatchToListt
applyScalet
applyRotationt appendt convertListToBatch( t imgst rotationt scale_mint scale_maxt flip_lvlt new_imgst rot_xt rot_yt rot_zt zoom_valt it lt
spline_orderst scaledt rotated( ( s utils/augmentations.pyt augmentPatchLossy s ))) c C s g } | j \ } } } } } xk t | j d D]V } | j | | d d
d d
d d
d d
f j d | | | | g q2 W| S( Ni i ( t shapeR R t reshape( t imgR t bt ct d1t d2t d3R ( ( s utils/augmentations.pyR ! s
Tc C s | d j \ } } } } } t j t | | | | | g } xO t t | D]; } | | | | d d
d d
d d
d d
f <qS W| S( Ni ( R R t zerosR R ( t img_listR R R R! R" t aR ( ( s utils/augmentations.pyR
( s
$9c C s5 g } t j j d d k } t j | d k d d } xó t t | D]ß } t | | } | d k r d g t | } n d g t | } t | | } t j j d t j d } t j j d t j d } t j j d t j d }
t
| | | |
g | } | j t | qN W| S( Ni g à?i i i i g @(
R R t randt whereR R R t applyFLIPS2t randintR R R R
( R R t pt locationsR R R t flippedR R R R ( ( s utils/augmentations.pyt augmentPatchLossLess/ s c C s t | } t | } | S( N( R t augmentPatchLessLess( R ( ( s utils/augmentations.pyt augmentBothD s c C s t j j | d | d t j d } t j j | d | d t j d } t j j | d | d t j d } | | | f S( Ni g f@i i ( R R R R ( R R R R ( ( s utils/augmentations.pyt getRotationValI s )))c C s t j j | | S( N( R R R ( R
R ( ( s utils/augmentations.pyt
getScalingValO s c C sª | d k r' t j j d d k } n t j j d d k } t j | d k d d } g } xA | D]9 } x# | D] } t j | d | } qv W| j | qi W| S( Ni i g à?i i t axis( R R R& R' t flipR ( t imagesR R* R+ R R R ( ( s utils/augmentations.pyt
applyFLIPSR s
c C sN g } xA | D]9 } x# | D] } t j | d | } q W| j | q
W| S( NR2 ( R R3 R ( R4 R+ R R R ( ( s utils/augmentations.pyR( ` s
c
C s$ t j d d d g d t j | d t j | d g d t j | d t j | d g g } t j t j | d d t j | d g d d d g t j | d d t j | d g g } t j t j | d t j | d d g t j | d t j | d d g d d d g g } t j | t j | | } g } xÁ t | D]³ \ } } d t j | j j d t j }
|
|
j t j
j | } t j
j d | j d | d
| d | | d d
} | t j t j d d
f } | j | qiW| S( Ng ð?g i i i g à?t dtypet inputt matrixt offsett ordert modet nearest( R t arrayt cost sint dott enumeratet asarrayt squeezeR t int64t linalgt invt scipyt ndimaget affine_transformt newaxisR (
R4 t rotR t transform_xt transform_yt transform_zt transformR R R t mid_indexR9 t new_img( ( s utils/augmentations.pyR h s. '/, 0-& %
c C s¦ g } x t | D] \ } } t j q t j d yU t j j | j | d | | } | t j t j d d
f } | j
| Wn n XWd QXq W| S( Nt ignoreR: ( RA t warningst catch_warningst simplefilterRG RH t zoomRC R RJ R ( R4 R R R R R RQ ( ( s utils/augmentations.pyR s
% ( t numpyR t nibabelt nibt
scipy.ndimageRG RS t PPt sysR R R
R- R/ R0 R1 R5 R( R R ( ( ( s utils/augmentations.pyt <module> s