a "DfG @s8ddlZddlmZGdddeZejdfddZdS)N) normalizec@s4eZdZdZd ddZeddZejfddZ dS) Nchannel2RGBzzConvert nchannel array to rgb by PCA. Parameters ---------- pca: sklearn.decomposition.PCA PCA. NcCs||_d|_dS)N)NN)_pca_min_max_value)selfpcar \/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imgviz/nchannel.py__init__szNchannel2RGB.__init__cCs|jS)zPCA for N channel to 3.)r)rr r r rszNchannel2RGB.pcac Csddl}|jdksJdt|jtjs2Jd|j\}}}|d|}|jdurt|j j ddd|_|j |}n |j |}|||d}|tj kr|jdurtj|d d tj|d d f|_|j\}} t||| }|d tj }nt|tjsJ||}|S) a\Convert nchannel array to rgb by PCA. Parameters ---------- nchannel: numpy.ndarray, (H, W, C), float N channel image. dtype: numpy.dtype Dtype (default: numpy.uint8). Returns ------- dst: numpy.ndarray, (H, W, 3), numpy.uint8 Visualized image. rNznchannel.ndim must be 3znchannel.dtype must be floatingi)Z n_componentsZ random_state)rr)Zaxis)Zsklearn.decompositionndimnpZ issubdtypedtypeZfloatingshapeZreshaper decompositionZPCAZ fit_transformZ transformuint8rZnanminZnanmaxrroundZastype) rnchannelrZsklearnHWDdstZ min_valueZ max_valuer r r __call__s2          zNchannel2RGB.__call__)N) __name__ __module__ __qualname____doc__r propertyrrrrr r r r rs   rcCst|||S)a`Convert nchannel array to rgb by PCA. Parameters ---------- nchannel: numpy.ndarray, (H, W, C), float N channel image. dtype: numpy.dtype Dtype (default: numpy.uint8). pca: sklearn.decomposition.PCA PCA. Returns ------- dst: numpy.ndarray, (H, W, 3), numpy.uint8 Visualized image. )r)rrrr r r nchannel2rgbKsr!)numpyrrobjectrrr!r r r r s E