Datasets
Models
Downloads: 1
31 lines (31 with data), 3.2 kB
3
` ã @ sh d dl Z d dlZd dlZd dlZd dlmZ d dlm Z dddZ
dd d
Z d d Z d
d Z
dd ZdS )é N)Úreslice)Ú AffineMapé Únearestç ô?c C s| t j| jj d d
d
}t jt j| j|kj sF| j| j| | j j
}| j d d
}| jj
} | jj d d
}
|d k r |||
d |d | f} n"|d k r´|||f} n|||
d f} t j| } xJt t
||
| D ]6\} \}
}}|
| | }|d dkrÞ| | d8 < qÞW t|| |
| ||d\}}tj||}|jj d d
}| }t j||ddsxt|| tjd |S ) Né r é g à?gü©ñÒMbP?)ÚorderÚmodeg{®Gáz?)Úrtolz/Output resolutions are different than expected!)r )ÚnpÚarrayÚheaderÚ get_zoomsÚabsÚdiagÚaffineÚallZ set_sformÚ get_fdataÚcopyÚshapeÚ enumerateÚzipr ÚnibÚ Nifti1ImageÚallcloseÚprintÚwarningsÚwarn)Úniftir r
Úin_plane_resolution_mmZslice_thickness_mmZnumber_of_slicesÚ
resolutionÚdatar r ÚzoomsZ new_zoomsÚiZn_iZres_iZ res_new_iZn_new_iZdata_resampledZaffine_resampledÚnifti_resampledÚxÚy© r( ú../src/utils/util.pyÚresample_nifti s>
r* c C s$ | | j |dd | j|ddd S )NT)ÚaxisÚkeepdimsg:0âyE>)ÚmeanÚstd)r& r+ r( r( r) Únormalize_zscore5 s r/ c C s4 t jj| }|jdd dkr0tj| }t|}t|jdkr¤|j\}}}|j |d d |d d
|d d |d d
f j
ddd}t |d d}|||fS t|jd kr0|j\}}}}|j |d d |d d
|d d |d d
f j
dddd}|j || d
d
f}t |d d}|||fS d S )
NÚ.r Úniir r é@ r )r+ é é )r r )r r )
ÚosÚpathÚbasenameÚsplitr Úloadr* Úlenr r Ú transposer/ Úreshape) Úfilenamer7 r r% ÚnxÚnyÚnzr& Únfr( r( r) Úload_from_file_CarSON9 s
B
D rB c
C s t jj| }|jdd d }t jjt jj| |}t|jdkrZ|j\}}} |jd }nBt|jdkr|jd d
\}}} }
|j |
| dddf}|jd }|S )
Nr0 r z_segmentation.nii.gzr r r r3 r4 r2 )r )r r r r )r r r r r3 )r5 r6 r7 r8 ÚjoinÚdirnamer: r r; r< r Úzerosr r r r* r r Ú to_filename)
r= r' Zx_niftiZx_nifti_resampledr7 Z basename_segZ filename_segr> r? r@ rA Zy_nifti_resampledZy_niftir( r( r) Úsave_to_file_nifti_CarSONP s"
rG c C s t jj| }|jdd dkrtj| }t|}|j\}}}}|j |d d |d d
|d d |d d
f j dddd}|S d S ) Nr0 r r1 r r2 r r )r+ ) r5 r6 r7 r8 r r9 r* r r r; r/ ) r= r7 r r% r> r? r@ rA r& r( r( r) Úload_from_file_CarMENj s
D rH )r r r NN)N)r5 r Únumpyr Znibabelr Zdipy.align.reslicer Zdipy.align.imaffiner r* r/ rB rG rH r( r( r( r) Ú<module> s
,