]i,ddlmZGddZdS))YOLOceZdZddZdZdS) EggDetector皙?c<t||_||_dS)N)rmodelconf_threshold)self model_pathr s O/data/users/s19104799/parasitic-detection.dcism.org/backend/src/egg_detector.py__init__zEggDetector.__init__s*%% ,c |||j}g}t|D]\}}|jt |jdkrt|jD]\}}|jd\}} } } |jd} t|j d} |jj | }t|t| t| t| f\}} } } d}td||z }td| |z }t|j| |z}t|j| |z}|||||f}||| | | f||||ft#| || d}|||S)N)confr )imagebbox bbox_padded confidence class_nameclass_id)rr enumerateboxeslenxyxycpunumpyrintclsnamesmaxminwidthheightcropfloatappend)r rresultscropped_imagesiresultjboxx1y1x2y2rrrpadding x1_padded y1_padded x2_padded y2_padded cropped_img cropped_infos r detect_eggszEggDetector.detect_eggs s**U)<*=="7++ 8 8IAv|'C ,=,=,A,A' 5588FAs%(Xa[__%6%6%<%<%>%>NBB!$!!2!2!8!8!:!:J"371:>>#3#3#9#9#;#;< >I #EL"w, ? ?I"'**"Iy)D##K "-!#RR 0(19i'S&+J&7&7&0$, $$L#)),7777rN)r)__name__ __module__ __qualname__r r9rr rrs7----$$$$$rrN) ultralyticsrrr=rr r?sI))))))))))r