Datasets
Models
Downloads: 1
152 lines (149 with data), 17.5 kB
B
%Yf÷I ã @ sÞ d Z ddlZddlZddlZddlmZ ddlmZ ddlZddlZddl m
Z ddl Z
ddlZddlZddlZddlmZmZ ddlmZ ddlmZ ddlmZm Z dd l m Z m Z m Z m!Z!m"Z" dd
l#m$Z$ e%e &d d ¡Z'ej(dCddi e )d¡ G dd dZ*e* Z+dedfddZ,dDddZ-dEd d Z.dFd d!Z/e dGd#d$Z0dHd&d'Z1d(d) Z2d*d+ Z3dId,d-Z4e d.ed%fd/d0Z5dddd1d2ed"fd3d4Z6dJd6d7Z7dKd9d:Z8dLd;d<Z9ed=d>d?d2d2d@fdAdBZ:dS )Mz
Plotting utils
é N)Úcopy)ÚPath)ÚImageÚ ImageDraw)Úgaussian_filter1d)Ú Annotator)Ú TryExceptÚthreaded)ÚLOGGERÚ
clip_boxesÚincrement_pathÚ xywh2xyxyÚ xyxy2xywh)ÚfitnessÚRANKéÿÿÿÿÚfontÚsizeé ÚAggc @ s* e Zd Zdd Zd ddZedd ZdS )
ÚColorsc s( d} fdd|D _ t j _d S )N)ÚFF3838ÚFF9D97ÚFF701FÚFFB21DÚCFD231Ú48F90AÚ92CC17Ú3DDB86Ú1A9334Ú00D4BBÚ2C99A8Ú00C2FFÚ344593Ú6473FFÚ0018ECÚ8438FFÚ520085ÚCB38FFÚFF95C8ÚFF37C7c s g | ]} d | ¡qS )ú#)Úhex2rgb)Ú.0Úc)Úself© úJ/home/iml/Desktop/Talha/yolov5_P2_ROI_HCM_Micro_1000_Github/utils/plots.pyú
<listcomp>' s z#Colors.__init__.<locals>.<listcomp>)ÚpaletteÚlenÚn)r/ Úhexsr0 )r/ r1 Ú__init__# s zColors.__init__Fc C s2 | j t|| j }|r.|d |d |d fS |S )Né é r )r3 Úintr5 )r/ ÚiÚbgrr. r0 r0 r1 Ú__call__* s zColors.__call__c s t fdddD S )Nc 3 s, | ]$}t d | d | d
dV qdS )r9 r8 é N)r: )r- r; )Úhr0 r1 ú <genexpr>0 s z!Colors.hex2rgb.<locals>.<genexpr>)r r8 é )Útuple)r? r0 )r? r1 r, . s zColors.hex2rgbN)F)Ú__name__Ú
__module__Ú __qualname__r7 r= Ú staticmethodr, r0 r0 r0 r1 r ! s
r é zruns/detect/expc C s d|krd|kr| j \}}}}|dkr|dkrt|d d } tj| d ¡ |dd}
t||}tjt |d ¡dd d
\} } |
¡ } tj d d d x@t |D ]4}
| |
|
|
¡ ¡ ¡ d
¡ | |
d¡ q¦W t d| d| d| d¡ tj| ddd t ¡ dS )zê
x: Features to be visualized
module_type: Module type
stage: Module stage within model
n: Maximum number of feature maps to plot
save_dir: Directory to save results
ÚDetectÚSegmentr9 Ú_z features.pngr )Údimé T)Ú tight_layoutg©?)ÚwspaceÚhspaceé ÚoffzSaving z... (ú/ú)i, Útight)ÚdpiÚ bbox_inchesN)ÚshapeÚstrÚtorchÚchunkÚcpuÚminÚpltÚsubplotsÚmathÚceilÚravelÚsubplots_adjustÚrangeÚimshowÚdetachÚnumpyÚsqueezeÚaxisr
ÚinfoÚsavefigÚclose)ÚxÚ module_typeÚstager5 Úsave_dirÚbatchÚchannelsÚheightÚwidthÚfÚblocksÚfigÚaxr; r0 r0 r1 Úfeature_visualization6 s"
" rx éd c C s t | ¡ | ¡ |¡t | ¡ | ¡ |¡ }}t | |||f¡\}}}t t | |¡d d|jd d ¡}t t ||¡d d|jd d ¡}t |||f ¡S )Nr9 r ) ÚnpÚlinspacer\ ÚmaxÚ histogram2dÚclipÚdigitizerW Úlog)rl Úyr5 ÚxedgesÚyedgesÚhistZxidxZyidxr0 r0 r1 Úhist2dU s
.$$r
éÜ éPà é c s: ddl m m} fdd}||||d\}}|||| S )Nr )ÚbutterÚfiltfiltc s d| }| | } ||dddS )Ng à?ÚlowF)ÚbtypeZanalogr0 )ÚcutoffÚfsÚorderZnyqZ
normal_cutoff)r r0 r1 Úbutter_lowpassb s z/butter_lowpass_filtfilt.<locals>.butter_lowpass)r )Ú scipy.signalr r )Údatar r r r r ÚbÚar0 )r r1 Úbutter_lowpass_filtfilt^ s r é, c C s g }xnt | D ]b\}}|d |
d d
f ¡ dd¡\}}}t |jd df|¡}| t ||t||fd¡¡ qW t |d¡ ¡ S )Né )rA r9 r9 r9 r )
Ú enumerater[ ÚsplitrY ÚfullrW ÚappendÚcatr rf ) ÚoutputÚmax_detÚtargetsr; ÚoÚboxÚconfÚclsÚjr0 r0 r1 Úoutput_to_targetk s &"r¥ ú
images.jpgc s t | tjr | ¡ ¡ ¡ } t |tjr4| ¡ ¡ }d}d}| j\}}} }
t||}t |d ¡ t
| d ¡dkr|| d9 } tj t | t |
dfdtj
d} xpt| D ]d\} }
| |krÀP t |
| t | | }}|
dd d¡}
|
| |||
|||
d d
f< q®W | t
| |
}|dk rpt || ¡} t ||
¡}
t | t fd
d |
| fD ¡} t | |
d }t| t|d
|d|d}xt| d D ]þ} t |
| t | | }}|j||||
|| gd dd d |r,|j|d |d gt|| jd d
dd t|dkr®||d d
df | k }t |d d
d d
f j }|d d
df d¡}|jd dk}|rd n|d d
df }|jd r |
¡ dkrî|dd g |
9 < |ddg | 9 < n|dk r ||9 }|dd g |7 < |ddg |7 < xt|j ¡ D ]p\}}|| }t |}|r^|| n|}|sv|| dkr8|r| n| d|| d} |j!|| |d q8W q®W |j" #|¡ d S ) Ni r> g à?r r9 éÿ é )Údtyper8 c 3 s | ]}t | V qd S )N)r: )r- rl )Únsr0 r1 r@ s z plot_images.<locals>.<genexpr>g{®Gáz?é
T)Ú
line_widthÚ font_sizeÚpilÚexample)r§ r§ r§ )rs r é( )éÜ r± r± )ÚtextÚ txt_colorr r: g)\Âõ(ð?g Ð?Ú z.1f)Úcolor)$Ú
isinstancerY ÚTensorr[ Úfloatrf rW r\ rz r` r| r r: Úuint8r Ú transposer_ Úcv2ÚresizerB r Úroundrc Ú rectangler² r Únamer4 r
ÚTÚastypeÚtolistÚcolorsÚ box_labelÚimÚsave) Úimagesr ÚpathsÚfnameÚnamesÚmax_sizeÚ max_subplotsÚbsrJ r? ÚwÚmosaicr; rÅ rl r Úscaler Ú annotatorÚtiÚboxesÚclassesÚlabelsr¢ r¤ r¡ r£ rµ Úlabelr0 )rª r1 Ú plot_imagesu sh
("&
""".
r× Ú c C s¨ t | t | } }g }x,t|D ] }| ¡ | | jd d ¡ q W tj|ddd t d¡ t d¡ t ¡ t
d|¡ t d¡ tj t
|d dd t ¡ d S )
Nr Úlrz.-ÚLR)rÖ ÚepochzLR.pngéÈ )rU )r rc Ústepr Ú param_groupsr] ÚplotÚxlabelÚylabelÚgridÚxlimÚylimrj r rk )Ú optimizerÚ schedulerÚepochsro r rJ r0 r0 r1 Úplot_lr_scheduler¶ s
rè c C sØ t jdt jd} t| d d
d d
f }|d d
df |d d
df }}tjddddd\}}|j||d d
dd | d ¡ tjd
dd tjddddd\}}|d j |d d |d j |d d tjddd d S )Nzval.txt)r© rA r r9 )r r T)ÚfigsizerM iX r« )ÚbinsÚcmaxÚcminÚequalz
hist2d.pngi, )rU r8 )é r )rê z
hist1d.pngrÜ )
rz ÚloadtxtÚfloat32r r] r^ r
Ú
set_aspectrj r )rl r¡ ÚcxÚcyrv rw r0 r0 r1 Ú plot_val_txtÇ s "
rô c C s´ t jdt jdj} ddddg}tjdddd d
\}}| ¡ }xdtd D ]X}|| j| | d | | ¡ d
d| |
¡ d
d || ¡ || || ¡ qFW tj
ddd d S )Nz targets.txt)r© z x targetsz y targetsz
width targetszheight targetsr8 )rL rL T)ré rM rA ry z.3gz +/- )rê rÖ z targets.jpgrÜ )rU )rz rï rð rÀ r] r^ ra rc r ÚmeanÚstdÚlegendÚ set_titlerj )rl Úsrv rw r; r0 r0 r1 Úplot_targets_txtØ s 6 rú c
C s | rt | jnt |}d}|r8tjdddddd ¡ }tjddddd\}}xt| d ¡D ]ö}tj|tj d
ddd d d
dgddj
} |d kr t | j d ¡nt
|¡}|rdddddddg}
x>td D ]2} || j|| | ddd
d || |
| ¡ qÌW | d ¡ d } |j| dd|
f | d d|
f d ddd
|j dd ¡ d d ¡d q`W |jd t
d!d"d#d$d%d&g¡ d'd(d)d*d+d,gd-dd
d.d/d0 |jd1d2 | t d3d4d¡¡ | d
d5¡ | d6d7¡ | d8¡ | d9¡ |jd:d; |d< }td=| d> tj |d?d@ d S )ANFr8 rA )r« r T)ré rM r9 )rL rA z
study*.txtr r¨ é rL é )r© ÚusecolsÚndminÚPÚRzmAP@.5z
mAP@.5:.95zt_preprocess (ms/img)zt_inference (ms/img)zt_NMS (ms/img)z.-)Ú linewidthÚ
markersizer g Y@Z study_coco_rØ ÚyoloÚYOLO)r r rÖ g @@éÑ é éa é: é# é gÍÌÌÌÌLA@g @D@g E@g ÀG@gÙH@g ÀI@zk.-g Ð?Z EfficientDet)r r ÚalpharÖ gÉ?)r é é< é9 é é7 zGPU Speed (ms/img)z COCO AP valz lower right)Úlocz study.pngzSaving z...i, )rU ) r Úparentr] r^ ra ÚsortedÚglobrz rï rð rÀ ÚarangerW Úarrayrc rß rø ÚargmaxÚstemÚreplacerâ Ú
set_yticksÚset_xlimÚset_ylimÚ
set_xlabelÚ
set_ylabelr÷ Úprintrj )
ÚfileÚdirrl ro Zplot2rw Zfig2Úax2rt r rù r; r¤ r0 r0 r1 Úplot_val_studyå sH $" *
r# r0 c
sÈ t d|d d¡ | d d
df | d d
dd
f ¡ }}t| ¡ d }tj| ¡ dddd gd
}tj|d d d
t ddt ddd t
j |d dd t
¡ t
d¡ t
jdddd dd ¡ }|d j|t d||d ¡d dd t t¡ fd d t|D W d Q R X |d d ¡ dt| k rFd k r~n n4|d tt|¡ |d jt | ¡ d!d"d# n|d d$¡ tj |dd|d ddd% tj |dd |d& ddd% d| d d
dd&
f< t | d d
dd
f d' | d d
dd
f< t! "tj#d(tj$d)d* ¡}x4| d d+
D ]$^} }
t% &|¡j'|
dt(| d, q(W |d )|¡ |d *d-¡ x2d.D ]*} x"d/D ]} || j+| ,d0¡ q|W qrW t
j |d dd t
d1¡ t
¡ d S )2NzPlotting labels to z
labels.jpgz... r r9 rl r rs rr )ÚcolumnsTÚautor é2 )rê gÍÌÌÌÌÌì?)Úpmax)ÚcornerÚ diag_kindÚkindÚdiag_kwsÚplot_kwszlabels_correlogram.jpgrÜ )rU Úsvgr8 )rL rL )ré rM g à?gé?)rê Úrwidthc s. g | ]&} d j | dd t|D ¡qS )r8 c S s g | ] }|d qS )r§ r0 )r- rl r0 r0 r1 r2 $ s z*plot_labels.<locals>.<listcomp>.<listcomp>)ÚpatchesÚ set_colorrà )r- r; )r r0 r1 r2 $ s z plot_labels.<locals>.<listcomp>Ú instancesé éZ r« )ÚrotationÚfontsizerÔ )rl r rw rê r' r¨ iÐ )iÐ iÐ r¨ )r© r§ iè )rs ÚoutlinerQ )r r9 r8 r¨ )ÚtopÚrightÚleftÚbottomFr )-r
ri rº r: r| ÚpdÚ DataFrameÚsnÚpairplotÚdictr] rj rk Ú
matplotlibÚuser^ ra r rz r{ Ú
contextlibÚsuppressÚ Exceptionrc r r4 Ú
set_xticksÚset_xticklabelsÚlistÚvaluesr Úhistplotr
r Ú fromarrayÚonesr¹ r ÚDrawr¾ rà rd rh ÚspinesÚ set_visible)
rÕ rÊ ro r. r Úncrl rw Úimgr£ r¡ r rù r0 )r r1 Ú plot_labels s@ *"
& ,
rQ r Fc sÀ ddl m} p dd tdD tj|| ¡ ¡ ¡ t| dd}t t||} t dt
| d }
t t
| |
¡|
¡\} } |
d kr| ¡ n| g} xt| D ]}
| |
||
¡ d
¡ ¡ d d ¡¡ | |
d
¡ |d k r ||
|d k rd ||
nd }| |
j|ddd q W t j|ddd t ¡ |r¼t d| ¡ |d k rt dd fdd|d |
D ¡ ¡ |d k r¼t dd fdd|d |
D ¡ ¡ |S ) Nr )Ú denormalizec S s g | ]}d | qS )Úclassr0 )r- r; r0 r0 r1 r2 D s z imshow_cls.<locals>.<listcomp>iè )rK rL g à?r9 )r9 r8 r g g ð?rQ u ârØ r7 )r5 Úverticalalignmenti, rT )rU rV zSaving z
True: r´ c 3 s | ]} | d V qdS )Ú3sNr0 )r- r; )rÊ r0 r1 r@ W s z imshow_cls.<locals>.<genexpr>z
Predicted:c 3 s | ]} | d V qdS )rU Nr0 )r- r; )rÊ r0 r1 r@ Y s ) Úutils.augmentationsrR rc rY rZ Úcloner[ r¸ r4 r\ r½ r] r^ r_ r` ra rd rg Úpermuterf r~ rh rø rj rk r
ri Újoin)rÅ rÕ ÚpredrÊ ÚnmaxÚverbosert rR ru r5 Úmrv rw r; rù r0 )rÊ r1 Ú
imshow_cls@ s. (,
*
*r^ úpath/to/evolve.csvc
C sl t | } t | ¡}dd |jD }|j}t|}t |¡}tj ddd t
j d#ddi t d | d
| d xÊt
|d d
D ]¶\}}|d d
d | f }|| } t d
d|d ¡ tj||t||ddddd tj| | ¡ ddd tj| d| dddid |d d kr t g ¡ t |d d | d qW | d ¡}tj|d d! t ¡ t d"| d S )$Nc S s g | ] }| ¡ qS r0 )Ústrip)r- rl r0 r0 r1 r2 a s z plot_evolve.<locals>.<listcomp>)r« rî T)ré rM r r rL zBest results from row z of ú:rû r r r9 r Úviridisgé?Únone)r. Úcmapr Ú
edgecolorszk+é )r z = z.3grü )Úfontdictr z>15z: z.pngrÜ )rU zSaved )r )r r; Úread_csvr$ rH r rz r r] Úfigurer@ Úrcr r ÚsubplotÚscatterr
rß r| ÚtitleÚyticksÚ with_suffixrj rk )
Ú
evolve_csvr Úkeysrl rt r¤ r; ÚkÚvÚmur0 r0 r1 Ú plot_evolve] s.
ru úpath/to/results.csvc C s | rt | jnt |}tjddddd\}}| ¡ }t| d¡}t|s^td| ¡ dx|D ]}yÄt
|¡}d d
|j D }|j
d d
d f } xtd dd
dddddddg
D ]r\}
} |j
d d
| f d¡} ||
j| | d|jddd ||
j| t| d
ddddd ||
j|| dd q¸W W qf tk
rj }
zt d | d |
¡ W d d }
~
X Y qfX qfW |d ¡ |j|d d d! t ¡ d S )"Nr8 r )rî r T)ré rM z results*.csvz No results.csv files found in z, nothing to plot.c S s g | ] }| ¡ qS r0 )r` )r- rl r0 r0 r1 r2 s z plot_results.<locals>.<listcomp>r r9 r¨ rA rL rü r« r rû r¸ Ú.)ÚmarkerrÖ r r )Úsigmara Úsmooth)rÖ r rî )r5 z Warning: Plotting error for z: z results.pngrÜ )rU )r r r] r^ ra rG r r4 ÚAssertionErrorÚresolver; rh r$ rH r rÁ rß r r rø rD r
ri r÷ rj rk )r r! ro rv rw Úfilesrt r rù rl r; r¤ r Úer0 r0 r1 Ú plot_resultsx s(
& , r c C sÌ t jdddddd ¡ }ddd d
d d d
g}tt| d¡}xdt|D ]V\}}ytj|ddj d d
dd
f } | j
d }
t | |rt ||
n|
¡} | d d
| f } | d | d ¡ } | | d< xt|D ]\}
}|
t
| k rZt
|rø|| n |j dd¡}|j| | |
d|ddd | ||
¡ | d¡ x(dD ]}|j| d¡ q>W qÒ| ¡ qÒW W qJ tk
r } ztd| d | W d d }~X Y qJX qJW |d ¡ t jt|d d d d S ) Nr8 rA )rî r T)ré rM r9 ÚImageszFree Storage (GB)zRAM Usage (GB)ZBatteryz dt_raw (ms)zdt_smooth (ms)zreal-world FPSz frames*.txt)rþ r3 iâÿÿÿr Zframes_rØ rw r )rx rÖ r r ztime (s))r7 r8 Fz Warning: Plotting error for z; zidetection_profile.pngrÜ )rU )r] r^ ra rG r r r rz rï rÀ rW r r\ r4 r r rß rø r rM rN ÚremoverD r r÷ rj )ÚstartÚstoprÕ ro rw rù r} Úfirt Úresultsr5 rl Útr; r rÖ Úsider~ r0 r0 r1 Úprofile_idetection s0
* r zim.jpggR¸
ëQð?r« Tc C s* t | ¡ dd¡} t| }|rR|d d
dd
f d¡d d¡|d d
dd
f< |d d
dd
f | | |d d
dd
f< t| ¡ } t| |j |t
| d t
| d
t
| d t
| d
d d |rÐdnd
f } |r&|j j d
d
d t
t| d ¡}
t | d
d d d
f ¡j|
ddd | S )Nr rA r8 r9 r )r r9 )r r¨ )r r )r r8 T)ÚparentsÚexist_okz.jpg.é_ )ÚqualityÚ subsampling)rY ÚtensorÚviewr r| Ú unsqueezer
Úlongr rW r: r ÚmkdirrX r ro r rJ rÆ ) ÚxyxyrÅ r ÚgainÚpadÚsquareÚBGRrÆ r Úcroprt r0 r0 r1 Ú save_one_box² s 4, D$r )r )ry )r r r )r )Nr¦ N)r rØ )rØ rØ N)r_ )rv rØ )r r r0 rØ );Ú__doc__rB r_ Úosr Úpathlibr r» r@ Úmatplotlib.pyplotÚpyplotr] rf rz Úpandasr; Úseabornr= rY ÚPILr r Zscipy.ndimage.filtersr Úultralytics.utils.plottingr Úutilsr r Ú
utils.generalr
r r r
r Ú
utils.metricsr r: Úgetenvr rj rA r rà rx r
r r¥ r× rè rô rú r# rQ r^ ru r r r r0 r0 r0 r1 Ú<module> sP
@
--