Datasets
Models
Downloads: 1
16 lines (16 with data), 1.3 kB
3 5Qľa˛ Ń @ s0 d dl Zd dlZd dlZd ddäZd dd äZdS )Ú NÚ Ú Ú Tc C sŠ t dt| ââ tj| â}tj|â}xxt| |âD ]j\}} |dkrTt|jdâ| â|ksr|dkr.t|jdâ| â|k r.x$t|âD ]} |j|â |j| â q|W q.W t |â t |â t t||ââ} t j | â t| Ä \}}t dt|ââ ||fS )NzBefore Resample datas:T┌_FzAfter 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 ˙-/home/ws/yanghan/Newcode/dataprocess/utils.py┌data_resample4seg s < r c C sÉ t dt| ââ tj| â}xZ| D ]R}t|jdâ| â}|dkrD||ksT|dkr ||kr xt|âD ]}|j|â q^W q W t dt|ââ tj |â |S )NzBefore Resample datas:r TFzAfter Resample datas:) 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 Tr! )r! r r T)┌numpy┌npr r r r r r r r ┌<module> s