From b95f635da281e902915080a14fa0f5bf2bbd12bd Mon Sep 17 00:00:00 2001 From: abhi12-ayalur Date: Tue, 27 Aug 2024 17:47:02 +0000 Subject: [PATCH] hotfix: yolov8 warmup --- object_detection/yolov8/test-image.jpg | Bin 0 -> 3801 bytes object_detection/yolov8/yolov8_model.py | 13 +++++++++++++ 2 files changed, 13 insertions(+) create mode 100644 object_detection/yolov8/test-image.jpg diff --git a/object_detection/yolov8/test-image.jpg b/object_detection/yolov8/test-image.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7fb175b34202dc68f6b76e87f22e0a7faa453dcd GIT binary patch literal 3801 zcmbtX2{hDe8~=|nW=67Q4Kz547Qn}2IPtT_#U=8oR*Z}|QxtYs$R<2C}d4g!<(i;UAl=mp&_;g^+OA8Idr_u5r+z1}Lqb7OI^u@Vu=1Y#FlIa<0G`EcgX zio*)r=9`BDBu>k|E?d?W#cHa&><~TkbYWa%)IDZg*H#YW9zQrE}FXPF$(qn!5GY-nQ1edclohb#P zR`*Eaj8X63Ts1a(G)Nu3(MQ+HtH=`2ABx=oLsVzXiQYJLk9u!AOEot^tzS@@F^*Iw@V{w?XhZ~m5Y7VbZp z{tqwVe}M-6TWD5$UaVjcl#?9_h5W_-`<2V%q7uQQ=75tJ?0XJ_o=%0+Z_vrUJ2I&DzAQOFJzZE?3f{d`I(OnyoyyPDeVndBICqa8j^LZ^TPr(D zlWk{{jnKc!3cVujj%08Xg8C)Qq#fNdrpXXagru1)l0+S_tOH4-C7U5W^5}>~^SvuccV52|9eT%UV4A`4%Ab3uuf7h}?M~zwB;sJP6b%=6ShK zbl*yYf=R=X5QC_Y5Wm6nFZdX8@M6GSNUiIDrs*IhX8d7;zawd?b(i6&U+*B2<}<;r zyYg0ifglTwBC3@xm|4juDlJcllX3)j+$+a(ifs}<3cfjF1W_>B%(7M(cqinu!brEG z2=>h&Ap~`|={SUqb{qFqj=+eGrs6J*z@nn^sKHG$}3c<*9bZC zHwTDVNf69nQY&hiV9WQ;k6ZN1A;IHz~;PMB>? zqetV1IJ4s zU}Qpls1F&#q7x5S5tPttX1;*i|j`)Mbm*>9d0^0<=cd11Xx<>MQqb{RI>SkR0U;Vd7lmaU@j zL_C0=@kA&h0tU?nlJlxV*WtGjwGQ(a5W>6Jts6?Z;STQq+j{;bo>dSpkG-E-2$Pa&K+ZH=Jx(OPH2{CPCCwWV&XI>EJs>@&mBZ0w zQX{z#m_US=b~6DN0Z$uvp8(?iUI%x!X^0~T%2{U{+tJRzRliWvISdoj7*X54%hZdw zYQ{4nHYv;)d|CIr*c(`Z7Zl|*WjtAS38^ZdV~GQebH$i`RI?4@D7Lp!J=VSj*S?8Y z#?UrXc1*t16*{|Ax_tV;Ij-QV9Gm=gZK)qMPu<);dToMEpLw*MQsO)@@BHjc)5}+( zm4o`a85Y+qUr!0HMlyl3N#9sJqGE$msteIcHRh>sb#%ZO>JHCRWPeYpuOCs8(pHF$ zi56c0^C^z=ShpS>9*w=#cECwPC|2SrrXHuTm{x#KtGKA&NJ5U;i!1(BDS7?3QfjKI z?AR3#$M|o+73-y=gR6+w)05`q-j`aqC#L!)<$Wk%CykZ5V$snW zbW)}J+HjYtW@jR?`i}nUiw7~>nKsD9Oku)>3cKkdoa}6hop)820^`58kCiohQ$n(s z)TmWa^Fn3fppz?I_WN1+qo=~IY>%sq%;GFNu_XYzRHzW`CsED+j2~ZvAt>S^jFVdk zSvEq1*;0!$fFoyNpSwKM&S)JHE>@MBm26a}oe%vb*uApzX$G16f=+MBPYk{0d^u@G zISF#{TCR7!1egQ`8uvdPL1JRK?-kM@ylgCa&mm;chTF8^vlpkldt?k;X#}-6Ar2UK zbb_Y=AI_MfO%`p@ih`v!l8jv2imKbYKXA90&8v2Z(lXM-SVB8JDECwDQBsnrmM}7$ zLPCV`QgXQ-3D8x3YqRgAQDBe+II2;PUByqQ4HAHV6qh}GLGGt9g$L+YgkmW;qFGl98F hzbbQ{e_if=i1~;gIv=zriPf0EUzqG&41bsm{{t54N2>q; literal 0 HcmV?d00001 diff --git a/object_detection/yolov8/yolov8_model.py b/object_detection/yolov8/yolov8_model.py index 03c3187..ee72270 100644 --- a/object_detection/yolov8/yolov8_model.py +++ b/object_detection/yolov8/yolov8_model.py @@ -47,6 +47,19 @@ def __setup__(self): self.current_world_classes = None self.current_world_fast_classes = None + test_file = sieve.File("test-image.jpg") + + print("Warming up inference...") + + self.__predict__(test_file) + self.__predict__(test_file, models="yolov8n") + self.__predict__(test_file, models="yolov8l-face") + self.__predict__(test_file, models="yolov8n-face") + self.__predict__(test_file, models="yolov8s-world") + self.__predict__(test_file, models="yolov8l-world") + + print("Inference warmed up.") + def __predict__( self, file: sieve.File,