Datasets
Models
Downloads: 1
48 lines (48 with data), 3.2 kB
U
ƯsbÆ ã @ sJ d dl Z ddlT ddlT d dlZd dlZd dlZdZdd Zd
dd ZdS ) é Né )Ú*c C s@ g }t | d(}t |¡}|D ]}| |d ¡ q W 5 Q R X |S )NÚrr )ÚopenÚcsvÚreaderÚappend)ÚfilenameÚlinesÚfZ csvreaderÚline© r
ú=/home/ws/yanghan/uncertainty-aware/dataprocess/dataprocess.pyÚreadCSV
s
r Útrainé@ c . C s g }g }| j D ]T}ttj | jdt| d ¡}ttj | jdt| d ¡} ||7 }|| 7 }qttj | jdt| jd d ¡}
ttj | jdt| jd d ¡} |dkrdg } |D ]B}
|
d¡d }| d¡}| |d d |d d |d
¡ qÂg }|
D ]D}
|
d¡d }| d¡}| |d d |d d |d
¡ qt
| j ¡}g }g }g }g }g }g }| D ]¬}|d }||kr|| j | d }| j
| d
}| j| d } | j | d } | j | d } | jd | d } | |¡ | |¡ | | ¡ | | ¡ | | ¡ | | ¡ q|g } g }!g }"g }#g }$g }%|D ]¬}&|&d }||krF| j |& d }| j
|& d
}| j|& d } | j |& d } | j |& d } | jd |& d } | |¡ |! |¡ |" | ¡ |$ | ¡ |# | ¡ |% | ¡ qFtd tdt| tdt| td tt||||||||d|}'tt| |!|"|#|$|%||d|}(|'|(fS |dkrþg } |D ]H}
|
d¡d }| d¡}| d|d d |d d |d
¡ qvg }|
D ]H}
|
d¡d }| d¡}| d|d d |d d |d
¡ qÈg })g }*g }+g },t
| j¡}-| D ]H}|d }||-kr2| j| d }| j| d }|) |¡ |* |¡ q2|D ]H}&|&d }||-kr| j|& d }| j|& d }|+ |¡ |, |¡ qtt|)|*||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Î
&&
(
*
.
.
r* )r r r r ) ÚrandomÚ
segdataloaderÚutilsr ÚglobZcv2Úfoldr r* r
r
r
r Ú<module> s