searchindex.js

Search.setIndex({docnames:["index","installation","models","notebooks/anchor-label-assignment-visualization","notebooks/comparison-between-yolort-vs-yolov5","notebooks/export-onnx-inference-onnxruntime","notebooks/export-relay-inference-tvm","notebooks/how-to-align-with-ultralytics-yolov5","notebooks/inference-pytorch-export-libtorch","notebooks/model-graph-visualization","notebooks/onnx-graphsurgeon-inference-tensorrt","what_is_yolort","yolov5"],envversion:{"sphinx.domains.c":2,"sphinx.domains.changeset":1,"sphinx.domains.citation":1,"sphinx.domains.cpp":4,"sphinx.domains.index":1,"sphinx.domains.javascript":2,"sphinx.domains.math":2,"sphinx.domains.python":3,"sphinx.domains.rst":2,"sphinx.domains.std":2,"sphinx.ext.intersphinx":1,"sphinx.ext.viewcode":1,nbsphinx:4,sphinx:56},filenames:["index.rst","installation.rst","models.rst","notebooks/anchor-label-assignment-visualization.ipynb","notebooks/comparison-between-yolort-vs-yolov5.ipynb","notebooks/export-onnx-inference-onnxruntime.ipynb","notebooks/export-relay-inference-tvm.ipynb","notebooks/how-to-align-with-ultralytics-yolov5.ipynb","notebooks/inference-pytorch-export-libtorch.ipynb","notebooks/model-graph-visualization.ipynb","notebooks/onnx-graphsurgeon-inference-tensorrt.ipynb","what_is_yolort.rst","yolov5.rst"],objects:{"yolort.models":[[2,0,1,"","YOLOv5"],[2,0,1,"","yolov5l"],[2,0,1,"","yolov5m"],[2,0,1,"","yolov5m6"],[2,0,1,"","yolov5n"],[2,0,1,"","yolov5n6"],[2,0,1,"","yolov5s"],[2,0,1,"","yolov5s6"],[2,0,1,"","yolov5ts"]],"yolort.v5":[[12,2,1,"","AutoShape"],[12,2,1,"","Bottleneck"],[12,2,1,"","BottleneckCSP"],[12,2,1,"","C3"],[12,2,1,"","C3TR"],[12,2,1,"","Concat"],[12,2,1,"","Contract"],[12,2,1,"","Conv"],[12,2,1,"","DWConv"],[12,2,1,"","Detect"],[12,2,1,"","Ensemble"],[12,2,1,"","Expand"],[12,2,1,"","Focus"],[12,2,1,"","GhostBottleneck"],[12,2,1,"","GhostConv"],[12,2,1,"","Model"],[12,2,1,"","SPP"],[12,2,1,"","SPPF"],[12,0,1,"","add_yolov5_context"],[12,0,1,"","attempt_download"],[12,0,1,"","focus_transform"],[12,0,1,"","get_yolov5_size"],[12,0,1,"","intersect_dicts"],[12,0,1,"","letterbox"],[12,0,1,"","load_yolov5_model"],[12,0,1,"","non_max_suppression"],[12,0,1,"","scale_coords"],[12,0,1,"","select_device"],[12,0,1,"","set_logging"],[12,0,1,"","space_to_depth"]],"yolort.v5.AutoShape":[[12,3,1,"","classes"],[12,3,1,"","conf"],[12,4,1,"","forward"],[12,3,1,"","iou"],[12,3,1,"","max_det"],[12,3,1,"","multi_label"]],"yolort.v5.Conv":[[12,4,1,"","fuseforward"]],"yolort.v5.Detect":[[12,3,1,"","onnx_dynamic"],[12,3,1,"","stride"]],"yolort.v5.Model":[[12,4,1,"","fuse"],[12,4,1,"","info"]],yolort:[[12,1,0,"-","v5"]]},objnames:{"0":["py","function","Python function"],"1":["py","module","Python module"],"2":["py","class","Python class"],"3":["py","attribute","Python attribute"],"4":["py","method","Python method"]},objtypes:{"0":"py:function","1":"py:module","2":"py:class","3":"py:attribute","4":"py:method"},terms:{"0":[0,1,2,3,4,5,7,8,9,10,12],"00000":[4,7],"00005":4,"00543":10,"00591":10,"00610352":10,"0081":10,"00976562":10,"0102295":10,"0151367":10,"02":10,"02582":[4,5,8],"03331":5,"03334":[4,8],"03424":4,"04":[5,10],"04047":4,"05023":[4,5,8],"054667m":10,"05475":4,"07":[5,8],"07544":10,"07642":10,"08":10,"08597":10,"08698":10,"08923":4,"09":[5,8,10],"0d03de6":12,"0x60000a817250":6,"1":[1,2,3,4,5,6,7,8,9,10,12],"10":[3,4,5,6,7,8,10],"100":[8,10],"1000":12,"1062":10,"1080ti":10,"1080x810":8,"11":[3,4,5,6,7,8,10],"113":6,"114":[2,12],"115":5,"11788800":10,"119":6,"12":[3,4,5,6,7,8,10],"1228800":6,"126960640":10,"128":[4,5,7,10],"1280":12,"12900":4,"13":[3,4,5,6,7,8,10,12],"13011862528":10,"13141":7,"133":7,"137":[4,5,7,10],"14":[3,4,5,7,8,10],"140":[4,5,7,10],"142":8,"14307":10,"14346":[4,5,8],"145":7,"147712":[4,5,7,10],"15":[3,4,5,7,8,10],"150":10,"151":10,"152":[4,5,7,10],"15259":[4,8],"15266":5,"156928":[4,5,7,10],"15872":7,"159":7,"16":[3,4,5,7,8,10,12],"164220":[4,5,7,10],"164608":[4,5,7,10],"167040":[4,5,7,10],"169904":10,"17":[3,4,5,7,8,10],"176":10,"1760":[4,5,7,10],"179":[6,8],"179328":[4,5,7,10],"18":[3,4,5,7,8,10],"180":[4,5,7,10],"18560":[4,5,7,10],"18579":10,"188":10,"19":[3,4,5,7,8,10],"192":[4,5,7,10],"1e":[4,5,6,8],"2":[3,4,5,6,7,8,9,10,12],"20":[3,4,5,7,8,10],"2017":5,"2022":10,"203904":[4,5,7,10],"2096":10,"21":[3,4,5,7,8,10],"21094":10,"2157":[4,5,6,7,8,9,10],"219":[4,5,8],"22":[3,4,5,7,8,10],"220":7,"221":7,"221568":[4,5,7,10],"224":10,"225":[4,5,8],"226":10,"228":7,"229":7,"22912":[4,5,7,10],"23":[4,5,7,10],"232":10,"23233":10,"234":10,"238":[4,5,7,10],"23831":10,"24":[3,4,5,7,10],"240":10,"241":[4,5,8],"24832":[4,5,7,10],"25":[0,4,5,7,10,12],"255":[4,6],"256":[4,5,7,10],"26":[4,5,7,10],"264":[4,5,7,10],"26947":7,"27":[4,5,7,10],"2779":10,"27846":[4,5,8],"2791":10,"2793":10,"2794":10,"27966":10,"2797":10,"28":[4,5,7,10],"2809":10,"2810":10,"2815":5,"282":[5,6],"28942":10,"29":[4,5,7,10],"29184":[4,5,7,10],"296448":[4,5,7,10],"298":10,"3":[3,4,5,6,7,8,9,10,12],"30":[4,5,7,10],"300":12,"301":[4,5,7,10],"30176":7,"302734":10,"30298":10,"303":[4,5,7,10],"31":[4,5,6,7,9,10],"310":8,"312":9,"313":9,"32":[2,4,5,7,8,10,12],"320":12,"3246940":[4,5,7,10],"329216":[4,5,7,10],"33":[4,5,7,10],"332160":[4,5,7,10],"333":[5,6,10],"33548":10,"34":[5,6,8],"345":[4,5,8],"34561":10,"347":10,"35":[2,4,5,7,10],"355":[4,5,7,10],"35855":10,"364":10,"36992":[4,5,7,10],"37":5,"3701":[5,6],"37329":10,"374":[4,5,8],"3751":6,"38":[4,5,7,10],"380":[4,5,7,10],"38255":10,"383":10,"3833":10,"384":[4,5,7,10],"386":[4,5,8],"387":[4,5,8,10],"389":10,"39":[4,5,6,7,9,10],"391":10,"392":10,"4":[2,3,4,5,6,7,8,9,10],"40":[4,5,7,10],"402":[5,6],"405":10,"409600":6,"40977":7,"41119":4,"41199":10,"412":8,"417":7,"42":10,"42114":10,"42178":[4,5,8],"436":[4,5,7,10],"439":7,"44":[4,5,7,10],"442880":[4,5,7,10],"44397":10,"445":[4,5,6,7,8,9,10],"45":[0,4,5,6,7,10,12],"45207":10,"45874":10,"46":[5,6,10],"4672":[4,5,7,10],"47":5,"47092":10,"477":10,"47729":[4,5,8],"4800":[4,5,7,10],"488159":10,"48898":5,"48911":[4,8],"48960768":10,"495":7,"49536":[4,5,7,10],"496":7,"4970":6,"4971":6,"4972":6,"4973":6,"4987":10,"4c":12,"5":[3,4,5,6,7,8,9,10,12],"50":[4,5,8,9,10],"501":8,"502045":10,"5081":10,"50839":10,"512x640":10,"514":7,"5165":10,"51723":4,"518":7,"51904":10,"52475":4,"53":10,"53088":10,"531":7,"537354":10,"538255":10,"54":6,"542":[4,5,7,10],"54393":[4,8],"54399":5,"546631":10,"546754":10,"54921":10,"55":[6,8],"570557":10,"58":10,"58034":4,"582153":10,"584":8,"58661":10,"59913":10,"6":[2,3,4,5,7,8,9,10,12],"6003":10,"6021":10,"6031":10,"6035":10,"6039":10,"6043":10,"6053":10,"60596":4,"6061":10,"60645":4,"6069":10,"60950":10,"61035":[4,8],"61041":5,"614":[5,8],"6145":10,"615":[4,5,7,10],"62674":10,"633881":10,"63422":4,"64":[4,5,7,8,10],"640":[2,4,5,6,7,8,10,12],"64430":7,"65":6,"65326":5,"65341":[4,8],"66479":10,"66721280":10,"668":[6,9],"669":6,"6749":10,"677":10,"678":[4,5,8],"68":[4,5,7,10],"7":[3,4,5,6,7,8,9,10],"70077":[4,5,8],"701":8,"70359":[4,7],"71":5,"711":10,"72515":10,"72589":[4,7],"72629":[4,5,8],"739":[4,5,7,10],"73984":[4,5,7,10],"740":[4,5,8],"741":4,"743408":10,"74496":[4,5,7,10],"75":4,"751m":10,"76599":10,"767":5,"77234":10,"77318":7,"77686":[4,8],"77689":5,"77780":7,"77881":4,"78528":10,"78537":10,"78729":4,"79":9,"792":[4,5,7,10],"8":[1,3,4,5,6,7,8,9,10],"80":[2,4,5,7,10,12],"80334":5,"80341":[4,8],"80370":10,"80621":[4,5,8],"808":10,"809":[4,5,8,10],"8096":10,"810":4,"811":[4,5,8],"81213":10,"81482":10,"8201":10,"82202":10,"83066":7,"8320":[4,5,7,10],"84":10,"84216":10,"84486":[4,5,8],"84495":[4,7],"84572":10,"851807":10,"86":[4,5,7,10],"86469":7,"866":10,"867676":10,"868048":10,"86893":7,"869":[4,5,8],"87330":7,"874":[4,5,8],"876":10,"88235":[4,7],"88238":[4,5,8],"88487":10,"889191":10,"89635":10,"897":[4,5,8],"898":10,"899":10,"9":[1,3,4,5,6,7,8,9,10,12],"900421":10,"90880":[4,5,7,10],"91071":5,"91077":[4,8],"91098":7,"91492":7,"92395":10,"924805":10,"925":[4,5,7,10],"93188":10,"939":10,"94":[4,5,7,10],"940805":10,"94446":10,"95":3,"95587":10,"96":[4,5,7,10],"97":9,"98145":10,"98258":10,"98694":7,"98871":7,"99":10,"99316":10,"99679":10,"999878":10,"\u00b5s":[5,8],"break":6,"byte":10,"case":[5,6,9,10],"class":[2,3,12],"default":[2,12],"do":[4,10],"export":[2,4,6,7,8],"float":[3,4,5,6,10,12],"function":[4,6,7,8,9,10,12],"import":[0,2,3,4,5,7,8,9,10],"int":[2,3,12],"long":[3,6],"new":10,"return":[2,4,5,6,7,8,9,10,12],"true":[0,2,3,5,6,7,8,9,10,12],"while":10,A:[5,6],And:[0,6,10],As:[4,10],At:4,For:[3,4,6,10,12],If:[2,5],In:[0,5,6,9,10],Its:4,No:10,One:[6,10],The:[0,2,3,4,5,6,7,8,9,10],Then:4,There:10,These:10,To:[1,4,5,6,9],_0:9,_1:9,_2:9,_:5,__torch__:6,__version__:[5,8,10],_flatten:5,_get_shape_onnx:[5,6],_jit_intern:[6,9],_recurs:8,_vf:[4,5,6,7,8,9,10],abl:4,ablat:10,about:0,abov:[1,9,10],acceler:5,access:9,accur:4,accuraci:4,achiev:5,acquir:3,act:12,activ:[10,12],actual:[3,10],adapt:12,add:[6,12],add_yolov5_context:12,adopt:[0,4,5],adv_index:6,advanc:5,after:[4,10],against:10,agnost:[4,7,12],ai:3,airplan:[2,3],algorithm:10,align:[0,4],all:[1,9,12],alloc:10,allow:3,alreadi:10,also:[2,4,6,7,8,9],alwai:8,an:[0,2,4,5,6,7,9,10],anchor:[0,5,6,10,12],anchor_gener:3,anchor_grid:[3,5,6,9,10],anchor_match_visu:3,anchor_threshold:3,anchor_util:[5,6,9,10],anchors_encod:3,anchors_per_lay:3,ani:[2,5,6,9,10],anno_path:3,annot:3,annotation_path:3,append:[3,4,6,7,8,10],appl:[2,3],appli:12,ar:[2,3,4,6,9,10,12],arang:[3,6],arch:2,architectur:2,argument:[4,5,6,7,8,9,10],argwher:6,aris:4,around:3,articl:6,as_tensor:[3,5,6,9,10],asnumpi:6,aspect:[2,4],assert:10,assert_allclos:[4,5,6,7,8],assert_clos:10,asset:[4,5,6,7,8,10],assign:[0,10],astyp:6,aten:[4,5,6,7,8,9,10],atol:[4,5,6,8],attach_idx:3,attempt:10,attempt_download:[0,4,5,7,10,12],attr:[4,5,6,7,8,9,10],attribut:10,augment:12,auto:[4,12],autoshap:[4,7,12],avx512:6,axi:6,b:12,back:4,backbon:[2,3],backend:4,background:2,background_class:10,backpack:[2,3],backward:6,ball:[2,3],banana:[2,3],bar:2,base:6,basebal:[2,3],bat:[2,3],batch_siz:[3,10,12],batched_nm:5,bc:3,bchw:12,bear:[2,3],becaus:10,bed:[2,3],been:[4,5,6,7,8,10],befor:2,begin:[6,10],behavior:2,behaviour:2,being:5,below:[5,10],bench:[2,3],best:6,better:6,between:0,bgr:12,bia:3,bicycl:[2,3],bin:3,bird:[2,3],blob:[3,4,5,6,7,8,9,10,12],block:10,blockassign:10,boat:[2,3],book:[2,3],bool:[2,12],both:[0,2,4],bottl:[2,3],bottleneck:12,bottleneckcsp:12,bound:6,boundari:3,bowl:[2,3],box:[2,3,4,5,7,8,9,10],box_cod:10,box_convert:3,box_head:[5,6,10],boxes_yolort:4,boxes_yolov5:4,broccoli:[2,3],bu:[0,2,3,4,5,6,7,8,10],build:[3,6,10],builder:10,c1:12,c2:12,c3:[4,5,7,10,12],c3tr:12,c:[2,5,10,12],cach:[5,10],cake:[2,3],call:[4,5,6,9,10],callmethod:6,can:[0,2,4,5,6,7,8,9,10],car:[2,3],carrot:[2,3],cast:[4,6,10],cat:[2,3],catalog:10,caus:[5,6,9,10],cc:6,ceil:6,cell:[2,3],center:3,cento:10,cfg:12,ch:12,ch_in:12,ch_out:12,chair:[2,3],chang:[2,5,6,9,10],channel:5,check:[2,5,10],check_model:5,checker:5,checkpoint:[0,2],checkpoint_path:[0,2,4,5,7,10,12],chi:9,cl:12,clamp:10,clamp_:3,classes_to_visit:9,classif:[0,2],clock:[2,3],clone:[5,6,9],cloud:10,co:[4,5,6,7,8,10],coco128_dirnam:3,coco128_path:3,coco:[1,2,3,4,5,7,8,10],coco_instance_category_nam:2,cocoapi:1,cocodetectiondatamodul:3,code:[0,5,6,9,10],color:[3,4,7,8,10,12],color_list:[3,4,7,8,10],color_rgb2bgr:3,com:[0,1,3,4,5,6,7,8,9,10,12],combin:5,common:[4,5,7,10],comparison:4,compat:6,complet:[4,6],complex:0,compos:5,comput:[5,10],compute_loss:3,concat:[4,5,7,10,12],concaten:6,conda:[4,5,7,8,10],conf:[4,7,12],conf_thr:12,confid:[4,7],consist:4,constant:[5,6,9,10],construct:[5,6,9],constructor:2,consum:[6,8],contain:[2,10],contextlib:3,contract:12,contrib_conv2d_nchwc:6,conv2d:6,conv2d_nchw:6,conv2d_nchwc:6,conv:[4,5,7,10,12],convert_bgr_to_rgb:4,convolut:12,coord:12,coordin:[3,4],copi:[5,6,9,10],copyright:6,corner:3,correctli:7,couch:[2,3],could:5,cover:3,cow:[2,3],cpp:[4,5,6,7,8,9,10],cpu:[3,4,5,6,7,8,10],creat:[5,6,9,10,12],createinferruntim:10,creation:10,crossstagepartialnetwork:12,csp:12,ctx:6,cu113:8,cubla:10,cublaslt:10,cuda:[4,7,8,10],cuda_device_ord:[3,4,5,7,8,10],cuda_vis:10,cuda_visible_devic:[3,4,5,7,8,10],cudnn:10,cup:[2,3],current:[4,6,10],custom:2,cv2:[3,4,5,6,7,8,10,12],cv2_imshow:[3,4,7,8,10],cvtcolor:3,cxcywh:3,d2h:10,d3e36fa:12,data:[3,4,10,12],data_path:3,datamodul:[3,9],dataset:[5,8],datatyp:10,deal:10,debug:6,decid:2,deeplearn:10,def:9,defin:[3,4,5,6,7,8,9,10],demo:[2,10],dens:6,depend:2,deploi:0,deprec:[6,8,9],depth:12,depth_multipl:12,descript:10,deserializecudaengin:10,design:0,desir:5,detach:[5,6,8,9],detail:[0,4,6,10],detect:[0,2,3,5,12],detection_box:10,detection_class:10,detection_scor:10,detections_per_img:10,determin:12,determinist:3,detr:0,dets_yolort:4,dets_yolov5:4,dev:[5,6,8],develop:1,devic:[3,4,5,6,7,8,9,10,12],dict1:12,dict2:12,dict:2,dictionari:[2,12],differ:[0,2,5,6,9,10],difficult:[0,3],dim:3,dimens:12,dine:[2,3],displai:2,divid:6,divis:2,doc:[4,10],docker:10,document:[3,4,5,6,7,8,9,10],doe:10,doesn:2,dog:[2,3],domain:10,donut:[2,3],dot1:9,dot2:9,down:10,download:[0,2,4,5,7,10],drier:[2,3],dtype:[3,4,5,6,7,9,10],due:6,dummi:8,dure:2,dwconv:12,dyn:6,dynam:[0,5],e:12,each:[2,5,6,8],easi:[4,10],edg:2,effici:10,efficientnms_trt:10,either:1,eleph:[2,3],elis:10,els:[3,4,6,7,8],emb:[0,5,10],embed:10,enabl:4,end:[4,10],enforc:3,engine_path:10,enqueu:10,ensembl:12,enumer:6,env:6,environ:[3,7,8,10],equal:6,error:[4,5,6,9,10],essenc:4,eval:[0,2,4,5,7,8,10],evalu:[1,2],even:10,everi:[5,6,9,10],exampl:[2,12],exclud:12,execut:[5,6,9,10],executioncontext:10,exist:[6,10],expand:12,expand_dim:6,expans:12,expect:[2,5],experi:[0,10],explan:10,export_friendli:2,export_onnx:5,export_script_nam:8,export_tensorrt_engin:10,f:[4,5,6,7,8,10],facebook:10,fail:10,fals:[2,4,5,7,10,12],fast:12,faster:0,fastest:5,fcakyon:12,featureextractor:3,feed:6,field:[2,5],file:[0,2,8,12],filenam:[8,10],fill:2,fill_color:2,filter:3,fire:[2,3],first:[4,5,8,10],fix:2,fixed_shap:[2,4,10],float32:[3,5,6,9,10],floattensor:2,floor:[5,6],focu:[0,12],focus_transform:12,folder:12,follow:[0,1,2,5],fork:[2,3],form:5,format:2,fortun:4,forward:[6,9,12],fp32:10,friendli:[0,2],frisbe:[2,3],from:[0,1,2,3,4,8,9,10,12],from_numpi:6,from_pytorch:6,frontend:6,fuse:12,fuseforward:12,g:[3,12],gain:[3,4,10,12],gather:5,geforc:10,gener:[6,10,12],get:[1,3,4,5,7,8,10],get_image_from_url:[5,6,7,8],get_model_size_mb:[6,8],get_trace_modul:[6,9],get_yolov5_s:12,getattr:[6,8],getlogg:10,gflop:[4,5,7,10],ghostbottleneck:12,ghostconv:12,giraff:[2,3],git:1,github:[0,1,3,4,5,6,7,8,9,10,12],githubusercont:[7,10],give:[5,6,9,10],glass:[2,3],glenn:12,global:10,glove:[2,3],go:3,good:[4,5,6,7,8,10],gpu:10,grab:12,gradient:[4,5,7,10],graghsurgeon:10,graph:[0,4,5,10],graphsurgeon:10,graphviz:9,greater:6,greater_equ:6,grid:[3,5,6,10],grid_assign:3,grid_ij:3,grid_wh:3,grid_xi:3,grid_xy_invers:3,gridspac:3,ground:2,group:12,gt:9,gtx:10,guess:10,guid:10,h2d:10,h:[2,3,5,6,12],ha:[4,5,6,7,8,10],hah:7,hair:[2,3],handbag:[2,3],handl:12,hardwar:5,hasattr:[6,9],hash_prefix:12,have:[2,4,6],head:3,head_output:[3,5,6,10],height:12,helper:5,here:4,highest:6,highli:6,hors:[2,3],host:10,hot:[2,3],how:0,howev:4,html:10,http:[0,1,3,4,5,6,7,8,9,10,12],hub:12,huggingfac:[4,5,6,7,8,10],human:5,hwc:12,hybrid:6,hybrid_nm:6,hydrant:[2,3],i:[3,5,6,10,12],idx:3,ignor:[4,5,6,7,8,9,10],im:12,imag:[0,2,9,12],image1:12,image2:12,image_root:3,image_util:[3,4,5,7,8,10],imagegrab:12,images_with_anchor:3,img0_shap:12,img1_shap:12,img:[5,6,8,9,12],img_h:[3,5,6,8],img_on:5,img_path:[0,4,10],img_raw:[3,4,7,8,10],img_siz:[4,5,7,10,12],img_sourc:[4,6,7,8,10],img_src1:5,img_src2:5,img_train:3,img_two:5,img_w:[3,5,6,8],implement:[0,4],imread:[4,10,12],imshow_scal:[3,4,7,8,10],in_fmt:3,in_siz:6,includ:[2,4,5,12],incorrect:[5,6,9,10],ind:3,index:[4,5,6,7,8,9,10],indic:[3,5],inds_jk:3,inds_lm:3,inds_on:3,infer:[0,2,10,12],inferenc:5,info:12,inform:[2,12],init:10,initi:5,inject:6,inplac:12,input0:6,input:[2,3,4,5,6,9,10,12],input_nam:6,input_sampl:10,input_shap:6,insid:4,instal:[0,5,6,9],instead:[2,9],instruct:[1,5],int32:[3,10],int64:[4,7,10],int64tensor:2,integr:0,intel:6,interfac:[4,10],intermedi:5,intermediate_featur:3,intern:[2,4,5,6,7,8,9,10],interpol:4,intersect:12,intersect_dict:12,introduc:4,introductori:6,intuit:0,invers:3,io:3,iou:[4,7,12],iou_thr:12,iou_threshold:10,ipynb:[3,4,5,6,7,8,9,10],ir:10,ir_visu:9,is_avail:[4,7,8,10],is_half:5,item:[6,8],iter:[3,5,6,9,10],its:[0,2],j:3,jit:[4,5,8],jk:3,jm:3,jocher:12,jpg:[0,2,4,5,6,7,8,10,12],just:10,k:[3,8,12],kei:12,kernel:12,keyboard:[2,3],kite:[2,3],knife:[2,3],kwarg:[2,4,5,6,7,8,9,10],l387:12,l416:12,l739:12,l754:12,l:3,label:[2,3,4,5,7,8,10,12],label_path:[4,7,10],labels_yolort:4,labels_yolov5:4,languag:5,laptop:[2,3],larg:6,larger:3,latenc:10,latest:1,layer:[4,5,7,10,12],layout_transform:6,lead:[5,6,9,10],left:3,len:[3,4,5,6,7,8,10],less:[6,10],let:[4,5,8,10],let_list:6,letlist:6,letterbox:[0,2,4,5,7,12],level:6,lib:[4,5,6,7,8,9,10],librari:0,libtorch:[0,4],light:[2,3],lightningdeprecationwarn:[6,8,9],lightningmodul:[6,8,9],like:0,linear:10,link:10,list:[0,2,3,5,12],lk:3,llvm:6,lm:3,load:[0,2,10],load_from_yolov5:[0,2,4,5,7,10],load_nam:[3,7,8],load_yolov5_model:[4,7,12],loaded_optimizer_states_dict:[6,8,9],local:[6,9,10],log:[6,10],logger:10,logits_decod:10,longer:5,look:[4,5,6,7,8,10],loop:[5,6,8],loss:[2,8],lot:10,love:0,lt:10,m1:6,m:[3,10],machin:6,maco:6,mai:[1,5,6],main:[3,4,5,6,7,8,9,10],mainli:5,maintain:[2,4],make:0,mani:10,manual_se:3,map:5,match:[2,3,12],matrix:5,max:[3,6,10],max_det:12,max_output_box:10,max_pool2d:6,max_siz:2,maximum:[2,12],mcpu:6,mean:[5,6,8,10],measur:5,mechan:[0,5],median:10,memori:[6,8,10],memusagechang:10,memusagesnapshot:10,memusagestat:10,meshgrid:[4,5,6,7,8,9,10],meta:10,meter:[2,3],metric:10,mib:10,microwav:[2,3],might:[5,6,9,10],min:[6,10],min_siz:2,miniconda3:6,minim:10,minimum:[2,6],mkl:6,mod:[6,8,9],mode:[2,4,5,10],model:[0,3,6,12],model_func:6,model_out:[5,8],model_path:[0,4,5,7,10],model_s:[6,8],model_script:8,model_yolort:[4,7],model_yolov5:[4,7],modul:[0,2,4,5,7,10],moment:4,more:[0,2,4,6,10],most:6,motorcycl:[2,3],mous:[2,3],ms:[5,6,8,10],much:10,multi_label:12,multipl:[5,12],multipli:[3,6],n:[2,4,5,7,8,10,12],name:[3,4,6,7,8,9,10,12],nativ:[4,5,6,7,8,9,10],nc:12,ndarrai:12,ndarray_s:6,nearest:[4,5,7,10],need:9,neg:[5,6],network:[3,10],neural:3,new_shap:[4,7,12],newlin:12,next:3,ngc:10,nivdia:10,nm:[0,4,6,7,10,12],nms_thresh:[4,5,7,10],nn:[2,4,5,6,7,10,12],nn_modul:8,no_grad:[4,5,6,7,8],node:10,non:[4,12],non_max_suppress:[4,6,7,12],none:[2,3,4,5,7,8,10,12],normal:3,note:6,notebook:[3,4,5,6,7,8,9,10],now:[0,4,10],np:[3,4,6,12],num_anchor:3,num_class:2,num_detect:10,num_lay:3,num_target:3,number:[2,3,5,6,9,10,12],numpi:[3,6,12],nvidia:10,nvinfer1:10,o:10,object:[0,3],obtain:[7,8],off:3,offici:[0,1,4,5,6],offset:3,omit:12,onc:[5,8,10],one:[2,4,10],ones:3,ones_lik:3,onli:[2,4,6,10],onnx2trt_util:10,onnx:[0,2,4],onnx_dynam:12,onnx_graphsurgeon:10,onnx_model:5,onnx_path:[5,10],onnxruntim:5,op:[3,5,10],open:12,opencv:[4,12],openmp:6,oper:[4,5,6,10],opset:[5,10],opset_vers:5,opt:[4,5,7,8,10],opt_level:6,option:[1,2,8,10,12],orang:[2,3],org:[6,10],origin:4,ort_help:5,ort_outs1:5,ort_outs2:5,os:[3,4,5,7,8,10],other:[5,6,9,10],otherwis:3,our:[0,4,6],out:[4,5,6,8,9,10],out_fmt:3,out_pytorch:5,out_script:8,output:[2,5,10],outsid:[3,10],oven:[2,3],over:[5,6,9,10],p:12,packag:[4,5,6,7,8,9,10],pad:[2,4,12],param:[4,5,6,7,10],paramet:[2,4,5,7,10,12],park:[2,3],parse_imag:3,parse_single_imag:[3,4],parser:10,part:4,pass:[2,4,5,6,7,8,9,10,12],passcontext:6,path:[3,12],pathlib:3,pci_bus_id:[3,4,5,7,8,10],peak:10,peel:4,per:[5,6,8,12],percentil:10,perform:[0,2,6,8,10],persist:10,person:[2,3],phone:[2,3],pil:[4,12],pip:[1,5,6,9,10,12],pipelin:[2,4,5,10],pizza:[2,3],plant:[2,3],pleas:[0,6,8],plot_one_box:[3,4,7,8,10],plugin:10,plugin_namespac:10,plugin_vers:10,pool:[6,12],posixpath:12,possibl:[0,2,12],post:[0,2,4,5,10],postprocess:9,pot:[2,3],power:[4,6],ppwwyyxx:1,pre:[0,4,5,8,10],prebuilt:10,precis:10,predict:[0,2,4,5,12],predictions2:2,predictions_pt:10,predictions_trt:10,predictorort:5,predictortrt:10,prepar:5,preprocess:[4,7,12],pretrain:[0,2,6,8,9],prim:6,principl:0,print:[4,5,6,7,8,9,10,12],printable_graph:5,prioriti:6,process:[0,2,4,5,8,10],prod:6,produc:[5,10],profil:[10,12],progress:2,project:9,proper:6,properti:[6,8,9],provid:[3,4,5,10],pt:[0,2,4,5,7,8,10,12],publish:10,py:[1,4,5,6,7,8,9,10,12],pycocotool:1,pypi:10,pyramid:12,python3:[4,5,6,7,8,9,10],python:10,pythonapi:1,pytorch2onnx:10,pytorch:[1,4,6,12],pytorch_lightn:[6,8],qp:10,quick:[5,6],r3:[2,12],r4:[2,12],r6:[2,7],racket:[2,3],rand:10,random:3,randrang:3,rang:[2,3,5,6,10],rather:[5,6,9],ratio:[2,3,4],ratio_pad:12,ratios_filt:3,ratios_wh:3,raw:[7,10],rcnn:[0,6],re:10,read:0,read_image_to_tensor:[4,5,7,8],readabl:5,reason:4,recommend:[5,6,9],rectangl:2,redirect_stdout:3,reduc:[6,10],refer:[6,9],refitt:10,refriger:[2,3],regist:[5,6,9,10],regress:2,rel:4,relai:[9,10],releas:[0,1,2,4,5,6,7,8,9,10,12],remot:[2,3],remov:[6,8,9],render:9,repeat:[3,6],repeat_interleav:3,repo:12,repositori:1,represent:5,reproduc:3,request:[4,7,8,10],requir:[0,1,2,4,5,6,7,8,9,10],requires_grad:6,requires_grad_:[5,6,9],rescal:[2,5,7],reshap:[5,6],resiz:[2,4,6],resize2d:6,resolv:[4,5,6,7,8,10],respons:[4,7,8,10],restor:4,result:[3,6,9,12],retinanet:0,return_lay:3,rgb:[5,12],robust:[4,12],round:6,routin:10,rt:[1,3,4,5,6,7,8,9,10],rtol:[4,5,6,8],run:[5,6,8,10,12],runtim:[0,4,6,10],s:[3,7,8,9,10,12],s_orig:9,safe:[5,6,9,10],safe_download:[4,10],same:[0,3,4,5,6,9,10,12],sandwich:[2,3],save:[8,10],scale:[4,10,12],scale_coord:[4,7,12],scale_factor:[5,6],scale_fil:12,scaleup:12,schedul:6,schema:10,scissor:[2,3],score:[2,4,5,7,8,10],score_activ:10,score_thresh:[0,2,4,5,7,8,10],score_threshold:[6,10],scores_yolort:4,scores_yolov5:4,scratch:10,screen:12,script:4,script_modul:6,scriptabl:4,seamlessli:0,search:10,second:10,see:[0,4,5,6,10],seed:3,select_devic:12,self:[5,6,9,10],sens:0,serial:10,set:2,set_input:6,set_log:12,setup:1,shape:[0,2,3,4,5,6,7,8,9,10,12],shape_list:6,shape_of:6,share:0,sheep:[2,3],shift:[3,5,6,9,10],shiftntopdown:10,shortcut:12,should:[0,2,4,5,6,10],sigmoid:[3,6],sign:[2,3],signific:0,silent:[5,6,9,10],simpl:0,simplifi:0,sinc:6,sink:[2,3],site:[4,5,6,7,8,9,10],size:[2,5,6,8,10,12],size_divis:[2,4,5,10],skateboard:[2,3],ski:[2,3],skip_val_set:3,skylak:6,slice_lik:6,slightli:3,slowest:5,snowboard:[2,3],so:[0,4,5],solut:[5,6],some:[4,5,10],sourc:[0,2,12],sourcetensor:[5,6,9],space:[4,5,6,7,8,10,12],space_to_depth:12,spatial:12,specifi:[2,12],speed:5,split:[4,6,7,8,10],spoon:[2,3],sport:[2,3],spp:12,sppf:[4,5,7,10,12],sqrt:6,squar:3,squeez:6,src:[4,5,6,7,8,9,10],stack:[1,3,4,5,6,7,8,9,10],standard:12,start:[5,6,8],std:[5,6,8],stderr:2,still:4,stop:[2,3],store:10,stori:10,str:[2,12],straightforward:5,strategi:[0,4],stride:[2,4,5,6,7,8,9,10,12],strided_slic:6,string:[2,10],stringio:3,strip:[4,7,8,10],structur:[0,4],subdirectori:1,subdivid:3,subsequ:[8,10],subtract:[3,6],success:10,successfulli:10,suitcas:[2,3],summari:[4,5,7,10],support:[0,2,4,6,10],suppress:12,surfboard:[2,3],sy:12,symbolic_opset9:5,system:5,t:[2,3,5,6,9,10],tabl:[2,3],take:6,target:[0,2,5,6,8],targets_append:3,targets_box:3,targets_cl:3,targets_train:3,targets_with_gain:3,task:0,te:6,teddi:[2,3],temporarili:12,tenni:[2,3],tensor:[2,3,4,5,6,7,8,9,10,12],tensorrt:0,tensorshap:[4,5,6,7,8,9,10],test:[5,8],test_dataload:3,text:[4,7,8,10],than:[0,5,9],thei:2,therefor:4,thi:[0,3,4,5,6,7,8,9,10],three:5,threshold:[4,7],through:9,throughput:10,tie:[2,3],time:[5,6,9,10],timeit:[5,6,8],to_numpi:5,toaster:[2,3],togeth:0,toilet:[2,3],tolist:[3,4,6,7,8,10],took:[5,10],tool:10,toothbrush:[2,3],torch:[2,3,4,5,6,7,8,9,10,12],torch_r:6,torchscript:4,torchscriptvisu:9,torchvis:[0,1,3,5,6],total:10,trace:[4,5,9,10],trace_wrapp:6,traceabl:4,tracerwarn:[5,6,9,10],tracewrapp:6,tracing_modul:9,traffic:[2,3],train2017:[2,3],train:[0,4],train_dataload:3,trainer:9,transform:[3,5,6,9],transpos:[5,6],trick:7,trigger:[4,5,6,7,8,9,10],trt:10,trt_helper:10,trtexec:10,truck:[2,3],truth:2,tune:6,tupl:[2,8,9,12],tupleconstruct:6,tupleunpack:6,tutori:[5,6],tv:[2,3],tvm:[0,2,4],tvm_re:6,two:3,type:[4,5,6,7,8,9,10],u:[1,9,10],ubuntu:10,uint8:4,ultralyt:[0,2,5,10,12],ultralytics_det:7,umbrella:[2,3],unfold:10,unlik:10,unstabl:6,unsupport:10,upcom:[4,5,6,7,8,9,10],upper:3,upsampl:[4,5,7,10],upstream:2,upstream_vers:2,uri:12,url:10,us:[1,2,5,6,8,9,10,12],usag:10,user:[6,9],userwarn:[4,5,6,7,8,9,10],usr:9,util:[0,3,4,6,7,8,10],v1:[6,8,9],v5:[0,4,5,7,10,12],v6:[0,4,5,7,10],v8201:10,v:8,valid:[4,6],valid_box:6,valu:[2,5,10,12],vanilla:4,vari:2,variabl:[5,6,9,10],variou:12,vase:[2,3],ve:10,verbos:[4,7,10,12],veri:[0,4],version:[1,2,4,5,6,7,10,12],vi:4,via:[5,6,10],view:[3,4,5,6,7,8,9,10],virtualmachin:[4,6],vision:6,visual:[0,4],vm_exec:6,vs:4,w:[2,3,5,10,12],wa:[5,6,8,10],wai:[4,5,10,12],walltim:10,want:10,warmup:10,warn:[5,6,8,9,10],we:[0,2,4,5,6,9,10],weight:0,well:[2,5],wh:[3,12],what:[0,3],wheel:10,when:5,where:[2,3,6],whether:2,which:[0,3,4,5],whole:[4,10],width:12,width_multipl:12,wine:[2,3],wise:12,within:0,without:10,won:[5,6,9,10],wongkinyiu:12,work:[5,10],workspac:10,would:[5,6,9,10],wrap:[2,10],wrapper:[4,12],x1:2,x2:2,x64:5,x86:6,x:[3,6,9,10,12],xy:3,xyxi:[3,12],y1:2,y2:2,y:[3,12],y_runtim:[5,10],yaml:12,yolo:[2,4,5,7,8,9,10],yolo_featur:3,yolo_modul:[6,8,9],yolohead:9,yolort:[3,9,12],yolort_det:7,yolotransform:[3,10],yolov5:[0,1,2,3,6,9,10],yolov5_darknet_pan_s_r60:2,yolov5l:2,yolov5m6:2,yolov5m:2,yolov5n6:[0,2,4,5,7,8,10],yolov5n:2,yolov5s6:[2,7,10],yolov5t:2,you:[0,1,4,5,6,9,10],your:[2,6,10],zebra:[2,3],zero:[6,12],zeros_lik:3,zhiqiang:6,zhiqwang:[1,3,4,5,6,7,8,9,10],zidan:[0,2,4,5,6,7,8,10,12],zip:[3,4,5,6,7,8,10],zoo:6},titles:["Welcome to yolort\u2019s documentation!","Install yolort","yolort.models","Visualize the anchor-target assignment","What is the difference between yolort and yolov5","Deploying yolort on ONNX Runtime","Deploying yolort on TVM","How to align with ultralytics yolov5","Intuition for yolort","Visualize model graph","Deploying yolort on TensorRT","<no title>","Modules and utils for YOLOv5"],titleterms:{"0":6,"6":6,"case":0,"do":6,"export":[5,10],"function":5,"import":6,align:7,an:8,anchor:3,anoth:5,api:0,approach:4,ar:5,assign:3,backend:[5,6,8],batch:3,between:[4,7,8,10],box:6,coco128:3,compar:8,compil:6,consist:5,dataload:3,dataset:3,definit:8,deploi:[5,6,10],deploy:0,detect:[4,7,8,10],differ:4,document:0,download:6,engin:10,environ:[4,5],extractor:3,featur:3,from:[5,6,7],get:[0,6],graph:[6,9],how:7,imag:[3,4,5,6,7,8,10],infer:[4,5,6,7,8],inferenc:[4,10],initi:8,instal:[1,10],instruct:10,intermedi:3,intuit:8,larger:6,libtorch:8,load:[4,5,6,7],model:[2,4,5,7,8,9,10],modul:12,obtain:3,onnx:[5,10],output:[6,7,8],pipelin:3,pre:[2,6],predict:10,prepar:[4,7,10],process:6,pytorch:[5,8,10],quick:0,read:8,refer:0,relai:6,result:[4,5,7,8,10],runtim:5,s:[0,4,5],sampl:3,score:6,script:8,set:5,setup:3,solut:0,start:0,state:0,stride:3,structur:2,target:3,tensorrt:10,test:[3,4,6,7,10],than:6,time:8,torchscript:8,trace:6,train:[2,3,5,6],tutori:0,tvm:6,ultralyt:[4,7],up:5,updat:7,us:[0,4],util:[5,12],verifi:[4,5,6,7,8,10],visual:[3,9],visualis:[7,8,10],visul:3,vm:6,weight:[2,4,7,10],welcom:0,what:4,whether:5,yolort:[0,1,2,4,5,6,7,8,10],yolov5:[4,5,7,8,12],yolov5n:6}})