h1 , d dl Z d dlZd dlmZmZ d dlZd dlZd dlmZ d dl m
Z
d dl m Z d dl
mZ g ej _ G d d Zed k( rDd dlZ eej* Zed
kD rej* d
ndZ ee j1 yy) N)AnyList)YOLO)LOGGER)check_requirements)GITHUB_ASSETS_STEMSc P e Zd ZdZdeddfdZd dZd dZd dZd d Z d d
Z
d d Z y)
Inferencea
A class to perform object detection, image classification, image segmentation and pose estimation inference.
This class provides functionalities for loading models, configuring settings, uploading video files, and performing
real-time inference using Streamlit and Ultralytics YOLO models.
Attributes:
st (module): Streamlit module for UI creation.
temp_dict (dict): Temporary dictionary to store the model path and other configuration.
model_path (str): Path to the loaded model.
model (YOLO): The YOLO model instance.
source (str): Selected video source (webcam or video file).
enable_trk (bool): Enable tracking option.
conf (float): Confidence threshold for detection.
iou (float): IoU threshold for non-maximum suppression.
org_frame (Any): Container for the original frame to be displayed.
ann_frame (Any): Container for the annotated frame to be displayed.
vid_file_name (str | int): Name of the uploaded video file or webcam index.
selected_ind (List[int]): List of selected class indices for detection.
Methods:
web_ui: Set up the Streamlit web interface with custom HTML elements.
sidebar: Configure the Streamlit sidebar for model and inference settings.
source_upload: Handle video file uploads through the Streamlit interface.
configure: Configure the model and load selected classes for inference.
inference: Perform real-time object detection inference.
Examples:
Create an Inference instance with a custom model
>>> inf = Inference(model="path/to/model.pt")
>>> inf.inference()
Create an Inference instance with default settings
>>> inf = Inference()
>>> inf.inference()
kwargsreturnNc j t d ddl}|| _ d| _ g | _ d| _ d| _ d| _ d| _ d| _ d| _
g | _ d| _ ddi|| _
d| _ | j d | j d | _ t j d| j y) z
Initialize the Inference class, checking Streamlit requirements and setting up the model path.
Args:
**kwargs (Any): Additional keyword arguments for model configuration.
zstreamlit>=1.29.0r NFg ?g?modelu Ultralytics Solutions: ✅ )r streamlitstsourceimg_file_names
enable_trkconfiou org_frame ann_frame
vid_file_name selected_indr temp_dict
model_pathr info)selfr r s g/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/ultralytics/solutions/streamlit_inference.py__init__zInference.__init__8 s ./ !')
!4262 >>' " . "nnW 5DO 1$..1ABC c d}d}d}| j j dd | j j |d | j j |d | j j |d y )
z=Set up the Streamlit web interface with custom HTML elements.z-zUltralytics YOLO Streamlit Application
u Experience real-time object detection on your webcam, videos, and images
with the power of Ultralytics YOLO! 🚀
zUltralytics Streamlit Appwide)
page_titlelayoutT)unsafe_allow_htmlN)r set_page_configmarkdown)r menu_style_cfgmain_title_cfg
sub_title_cfgs r web_uizInference.web_uiU su Lv?
+Fv V 4@ 4@ $?r c f | j j 5 d}| j j |d ddd | j j j d | j j j dd | _ | j
dv r.| j j j
d d
d k( | _ t | j j j d d
d| j d | _
t | j j j dd
d| j d | _ | j
dk7 rI| j j d \ }}|j | _ |j | _ yy# 1 sw Y qxY w)zAConfigure the Streamlit sidebar for model and inference settings.z`https://raw.githubusercontent.com/ultralytics/assets/main/logo/Ultralytics_Logotype_Original.svg )widthNzUser ConfigurationSource)webcamvideoimage)r0 r1 zEnable Tracking)YesNor3 zConfidence Thresholdg g ?g{Gz?z
IoU Thresholdr2 )r sidebarr2 title selectboxr radior floatsliderr r columnsemptyr r )r logocol1col2s r r6 zInference.sidebarh sE
WW__ +uD GGMM$cM * +
23ggoo// (
;;- - "ggoo 3 34E} UY^ ^DO GGOO " "#93TYYPT U
//c488UYZ[ ;;' ! +JD$ !ZZ\DN !ZZ\DN "! + +s F&&F0c T ddl m}m} d| _ | j dk( r| j
j j d| }|_t j |j }t dd 5 }|j |j ddd d| _ yy| j d
k( rd| _ y| j d k( rddl
}| j
j j d |d
}|r|D ] }|j dd|j j! d d 5 } | j |j | j" j% | j |j d ddd yyy# 1 sw Y xY w# 1 sw Y xY w)z:Handle video file uploads through the Streamlit interface.r ) IMG_FORMATS VID_FORMATS r1 zUpload Video File)typeNzultralytics.mp4wbr0 r2 zUpload Image FilesT)rE accept_multiple_filesF.)deletesuffix)pathname)ultralytics.data.utilsrB rC r r r r6
file_uploaderioBytesIOreadopenwritetempfileNamedTemporaryFilerM splitr append)
r rB rC vid_filegoutrU imgfilesimgfiletfs
r
source_uploadzInference.source_upload s} C ;;' !ww445H{4[H#JJx}}/+T2 (cIIaffh'(%6" $
[[H
$!"D
[[G
# ww445IP[sw4xH ' \G!44EAgllN`N`adNefhNiMjJk4l \pr 0 ++22BGGW\\3Z[\ \\ $ ( (\ \s 7 F3AF F F' c g dg dct t D cg c], t fdD rdvrj dd . c}fd }| j r/|j d | j j
d
d d | j j j d | }| j j d
5 t |j d
| _
t | j j j! }ddd | j j# d | j j j% d|dd }|D cg c] }|j' | c}| _ t+ | j( t st | j( | _ yyc c}w # 1 sw Y xY wc c}w )zz&Inference.configure.. s 61q||A6s grayscaleyolor c j | d d j j | dd j xs d fS )N rD )indexlower)rj M_ORDT_ORDs r z%Inference.configure.. s< 5;;q!u{{}5u{{1QR5;;=CVTV7WX r )keyr z.pt ModelzModel is downloading...NzModel loaded successfully!Classes )default)sortedr anyreplacer insertrW r r6 r8 spinnerr rq r listnamesvaluessuccess multiselectrp r
isinstance) r rj available_modelsselected_model class_namesselected_classesoptionrr rs s ` @@r configurezInference.configure s OPu u! -
666;a;O &&)
Y
?? # #At'<'
FG,,.!$ GGOO$wx ?? "jj . . DII488TEVEV`d / G #jjTYYDHHVZVgVgj hG")!*//"3 KKMGGLLN$$UUDT$U$$_uN_$`+ ,,..
KKM r )r N) __name__
__module__ __qualname____doc__r r r+ r6 r_ r r r r r r
r
s? #JD D D:@&*.\088D*- r r
__main__rv )r )rP r typingr r r torch ultralyticsr ultralytics.utilsr ultralytics.utils.checksr ultralytics.utils.downloadsr r __path__r
r syslenargvargsr r r r r r s
$ 7 ;
c c L z sxx=D !8CHHQKE
E$$& r