Datasets
Models
Downloads: 1
55 lines (55 with data), 3.2 kB
B
`ÉaÆ ã @ sJ d dl Z ddlT ddlT d dlZd dlZd dlZdZdd Zd
dd ZdS ) é Né )Ú*c C sD g }t | d,}t |¡}x|D ]}| |d ¡ q W W d Q R X |S )NÚrr )ÚopenÚcsvÚreaderÚappend)ÚfilenameÚlinesÚfZ csvreaderÚline© r
ú@/home/wangqiuli/Code/Newcode1222_monk/dataprocess/dataprocess.pyÚreadCSV
s
r Útrainé@ c . C s& g }g }x^| j D ]T}ttj | jdt| d ¡}ttj | jdt| d ¡} ||7 }|| 7 }qW ttj | jdt| jd d ¡}
ttj | jdt| jd d ¡} |dkrxg } xJ|D ]B}
|
d¡d }| d¡}| |d d |d d |d
¡ qÈW g }xL|
D ]D}
|
d¡d }| d¡}| |d d |d d |d
¡ qW t
| j ¡}g }g }g }g }g }g }x´| D ]¬}|d }||kr| j | d }| j
| d
}| j| d } | j | d } | j | d } | jd | d } | |¡ | |¡ | | ¡ | | ¡ | | ¡ | | ¡ qW g } g }!g }"g }#g }$g }%x´|D ]¬}&|&d }||krX| j |& d }| j
|& d
}| j|& d } | j |& d } | j |& d } | jd |& d } | |¡ |! |¡ |" | ¡ |$ | ¡ |# | ¡ |% | ¡ qXW td tdt| tdt| td tt||||||||d|}'tt| |!|"|#|$|%||d|}(|'|(fS |dkr"g } xP|D ]H}
|
d¡d }| d¡}| d|d d |d d |d
¡ qW g }xP|
D ]H}
|
d¡d }| d¡}| d|d d |d d |d
¡ qâW g })g }*g }+g },t
| j¡}-xP| D ]H}|d }||-krP| j| d }| j| d }|) |¡ |* |¡ qPW xP|D ]H}&|&d }||-kr¢| j|& d }| j|& d }|+ |¡ |, |¡ q¢W tt|)|*||d|}'tt|+|,||d|}(|'|(fS d S )NZ data_foldz.csvZ mask_foldr r ú/éÿÿÿÿÚ_r é z.pngz.npyz _lung.npyz_mediastinal.npyz_red.pngz _blue.pngZmid_z _mask.pngz ***********zthe length of train data: zthe length of test data: z -----------)ÚwidthÚheightÚrow)Ztraining_fold_indexr ÚosÚpathÚjoinZcsvPathÚstrZtest_fold_indexÚsplitr ÚlistdirZ maskPath2ZmidPathZlungPathZ mediaPathZ maskPath1ÚprintÚlenÚloaderÚDatasetZrowPathZ
RowDataset).ÚconfigÚmodeÚ batchsizer r Z train_datasZ train_masksÚindexZtempdataZtempmaskZ
test_datasZ
test_masksZtemp_train_datasÚoneZone_tempZone_listZtemp_test_datasZ mid_filesZtemp2_train_interZtemp2_train_unionZtemp2_train_dataZtemp2_train_lungZtemp2_train_mediaZtemp2_train_maskZone_train_dataZ imagenameZ
innertemp0Z
innertemp1Z
innertemp2Z
innertemp3Z
innertemp4Z
innertemp5Ztemp2_test_dataZtemp2_test_lungZtemp2_test_mediaZtemp2_test_interZtemp2_test_unionZtemp2_test_maskZ
one_test_dataÚ
dataloaderÚdataloader_valZtemp2_train_datasZtemp2_train_masksZtemp2_test_datasZtemp2_test_masksZ row_filesr
r
r Úget_dataloader sÎ &&
*
,
0
0
r* )r r r r ) ÚrandomÚ
segdataloaderÚutilsr ÚglobÚcv2Úfoldr r* r
r
r
r Ú<module> s