Datasets
Models
Downloads: 1
18 lines (18 with data), 1.3 kB
U
Ưsbò ã @ s0 d dl Zd dlZd dlZd
ddZd dd ZdS ) é Néþÿÿÿé é Tc C sÞ t dt| t | ¡}t |¡}t| |D ]f\}} |dkrRt| d¡| |ksp|dkr,t| d¡| |k r,t|D ]}
| |¡ | | ¡ qxq,t | t | t
t||} t | ¡ t| \}}t dt| ||fS )NúBefore Resample datas:TÚ_FúAfter Resample datas:)
ÚprintÚlenÚcopyÚdeepcopyÚzipÚintÚsplitÚrangeÚappendÚsortedÚlistÚrandomÚshuffle) Ú train_datasÚ train_masksÚindexÚ
splitvalueÚ
resample_rateÚis_bigZ res_datasZ res_masksÚ
train_dataZ
train_maskÚiÚtemp© r ú7/home/ws/yanghan/uncertainty-aware/dataprocess/utils.pyÚdata_resample4seg s
<
r c C s t dt| t | ¡}| D ]N}t| d¡| }|dkrB||ksR|dkr ||kr t|D ]}| |¡ qZq t dt| t |¡ |S )Nr r TFr )
r r r
r r
r r r r r ) r r r r r Zresample_datasr Úvaluer r r r Údata_resample4cls s
r" )r r r T)r r r T)ÚnumpyÚnpr r
r r" r r r r Ú<module> s