a "ºDfÚ ã@s$ddlZdd„Zdd„Zdd„ZdS)éNc Csâd}d}d}d}d}d}||||||}t |df¡}d}d|d|…df<t dt d|¡|¡|d|…d f<||}dt dt d|¡|¡||||…df<d||||…d f<||}d||||…d f<t dt d|¡|¡||||…d f<||}dt dt |¡|¡||||…d f<d||||…d f<||}d||||…d f<t dt d|¡|¡||||…df<||}dt dt |¡|¡||||…d f<d||||…df<|S)Nééééé éréÿéé)ÚnpÚzerosÚfloorZarange) ZRYZYGZGCZCBZBMZMRÚncolsÚ colorwheelÚcol©rúX/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imgviz/flow.pyÚmake_colorwheel s6&.*,*,rcCs\|jdd…\}}t ||dftj¡}tƒ}|jd}t t |¡t |¡¡}t ||¡tj}|dd|dd} t | ¡ tj¡} | d}d|||k<| | }t|jdƒD]œ} |dd…| f}|| d}||d}d||||}|dk}d||d||||<||d||<| }t d|¡|dd…dd…|f<qº|S)Nr rrr gào@gè?r) ÚshaperrZuint8rÚsqrtÚsquareZarctan2Úpir ZastypeÚint32Úrange)Úflow_uÚflow_vÚHÚWZ flow_imagerrÚradÚaZfkZk0Zk1ÚfÚiÚtmpZcol0Zcol1rÚidxZch_idxrrrÚflow_compute_color/s, "r$cCs¶|jdksJdƒ‚|jddks(Jdƒ‚t |jtj¡s@Jdƒ‚|dd…dd…df}|dd…dd…df}t t |¡t |¡¡}t |¡}d }|||}|||}t ||ƒS) zÂVisualize optical flow. Parameters ---------- flow_uv: numpy.ndarray, (H, W, 2), float Optical flow. Returns ------- dst: numpy.ndarray RGB image. rzflow must be 3 dimensionalr zflow must have shape (H, W, 2)zfloat must be float typeNrr gñhãˆµøä>) ÚndimrrZ issubdtypeZdtypeZfloatingrrÚmaxr$)Zflow_uvrrrZrad_maxÚepsilonrrrÚflow2rgbOs r()Únumpyrrr$r(rrrrÚs&