ultralytics 8.0.98 add Baidu RT-DETR models (#2527)
Co-authored-by: Kalen Michael <kalenmike@gmail.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com> Co-authored-by: Dowon <ks2515@naver.com>
This commit is contained in:
Glenn Jocher
GitHub
@ -1,173 +0,0 @@
|
||||
---
|
||||
description: Explore Ultralytics neural network modules for convolution, attention, detection, pose, and classification in PyTorch.
|
||||
---
|
||||
|
||||
# Conv
|
||||
---
|
||||
:::ultralytics.nn.modules.Conv
|
||||
<br><br>
|
||||
|
||||
# DWConv
|
||||
---
|
||||
:::ultralytics.nn.modules.DWConv
|
||||
<br><br>
|
||||
|
||||
# DWConvTranspose2d
|
||||
---
|
||||
:::ultralytics.nn.modules.DWConvTranspose2d
|
||||
<br><br>
|
||||
|
||||
# ConvTranspose
|
||||
---
|
||||
:::ultralytics.nn.modules.ConvTranspose
|
||||
<br><br>
|
||||
|
||||
# DFL
|
||||
---
|
||||
:::ultralytics.nn.modules.DFL
|
||||
<br><br>
|
||||
|
||||
# TransformerLayer
|
||||
---
|
||||
:::ultralytics.nn.modules.TransformerLayer
|
||||
<br><br>
|
||||
|
||||
# TransformerBlock
|
||||
---
|
||||
:::ultralytics.nn.modules.TransformerBlock
|
||||
<br><br>
|
||||
|
||||
# Bottleneck
|
||||
---
|
||||
:::ultralytics.nn.modules.Bottleneck
|
||||
<br><br>
|
||||
|
||||
# BottleneckCSP
|
||||
---
|
||||
:::ultralytics.nn.modules.BottleneckCSP
|
||||
<br><br>
|
||||
|
||||
# C3
|
||||
---
|
||||
:::ultralytics.nn.modules.C3
|
||||
<br><br>
|
||||
|
||||
# C2
|
||||
---
|
||||
:::ultralytics.nn.modules.C2
|
||||
<br><br>
|
||||
|
||||
# C2f
|
||||
---
|
||||
:::ultralytics.nn.modules.C2f
|
||||
<br><br>
|
||||
|
||||
# ChannelAttention
|
||||
---
|
||||
:::ultralytics.nn.modules.ChannelAttention
|
||||
<br><br>
|
||||
|
||||
# SpatialAttention
|
||||
---
|
||||
:::ultralytics.nn.modules.SpatialAttention
|
||||
<br><br>
|
||||
|
||||
# CBAM
|
||||
---
|
||||
:::ultralytics.nn.modules.CBAM
|
||||
<br><br>
|
||||
|
||||
# C1
|
||||
---
|
||||
:::ultralytics.nn.modules.C1
|
||||
<br><br>
|
||||
|
||||
# C3x
|
||||
---
|
||||
:::ultralytics.nn.modules.C3x
|
||||
<br><br>
|
||||
|
||||
# C3TR
|
||||
---
|
||||
:::ultralytics.nn.modules.C3TR
|
||||
<br><br>
|
||||
|
||||
# C3Ghost
|
||||
---
|
||||
:::ultralytics.nn.modules.C3Ghost
|
||||
<br><br>
|
||||
|
||||
# SPP
|
||||
---
|
||||
:::ultralytics.nn.modules.SPP
|
||||
<br><br>
|
||||
|
||||
# SPPF
|
||||
---
|
||||
:::ultralytics.nn.modules.SPPF
|
||||
<br><br>
|
||||
|
||||
# Focus
|
||||
---
|
||||
:::ultralytics.nn.modules.Focus
|
||||
<br><br>
|
||||
|
||||
# GhostConv
|
||||
---
|
||||
:::ultralytics.nn.modules.GhostConv
|
||||
<br><br>
|
||||
|
||||
# GhostBottleneck
|
||||
---
|
||||
:::ultralytics.nn.modules.GhostBottleneck
|
||||
<br><br>
|
||||
|
||||
# Concat
|
||||
---
|
||||
:::ultralytics.nn.modules.Concat
|
||||
<br><br>
|
||||
|
||||
# Proto
|
||||
---
|
||||
:::ultralytics.nn.modules.Proto
|
||||
<br><br>
|
||||
|
||||
# Ensemble
|
||||
---
|
||||
:::ultralytics.nn.modules.Ensemble
|
||||
<br><br>
|
||||
|
||||
# Detect
|
||||
---
|
||||
:::ultralytics.nn.modules.Detect
|
||||
<br><br>
|
||||
|
||||
# MLPBlock
|
||||
---
|
||||
:::ultralytics.nn.modules.MLPBlock
|
||||
<br><br>
|
||||
|
||||
# LayerNorm2d
|
||||
---
|
||||
:::ultralytics.nn.modules.LayerNorm2d
|
||||
<br><br>
|
||||
|
||||
# Segment
|
||||
---
|
||||
:::ultralytics.nn.modules.Segment
|
||||
<br><br>
|
||||
|
||||
# Pose
|
||||
---
|
||||
:::ultralytics.nn.modules.Pose
|
||||
<br><br>
|
||||
|
||||
# Classify
|
||||
---
|
||||
:::ultralytics.nn.modules.Classify
|
||||
<br><br>
|
||||
|
||||
# autopad
|
||||
---
|
||||
:::ultralytics.nn.modules.autopad
|
||||
<br><br>
|
||||
84
docs/reference/nn/modules/block.md
Normal file
84
docs/reference/nn/modules/block.md
Normal file
@ -0,0 +1,84 @@
|
||||
# DFL
|
||||
---
|
||||
:::ultralytics.nn.modules.block.DFL
|
||||
<br><br>
|
||||
|
||||
# Proto
|
||||
---
|
||||
:::ultralytics.nn.modules.block.Proto
|
||||
<br><br>
|
||||
|
||||
# HGStem
|
||||
---
|
||||
:::ultralytics.nn.modules.block.HGStem
|
||||
<br><br>
|
||||
|
||||
# HGBlock
|
||||
---
|
||||
:::ultralytics.nn.modules.block.HGBlock
|
||||
<br><br>
|
||||
|
||||
# SPP
|
||||
---
|
||||
:::ultralytics.nn.modules.block.SPP
|
||||
<br><br>
|
||||
|
||||
# SPPF
|
||||
---
|
||||
:::ultralytics.nn.modules.block.SPPF
|
||||
<br><br>
|
||||
|
||||
# C1
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C1
|
||||
<br><br>
|
||||
|
||||
# C2
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C2
|
||||
<br><br>
|
||||
|
||||
# C2f
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C2f
|
||||
<br><br>
|
||||
|
||||
# C3
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C3
|
||||
<br><br>
|
||||
|
||||
# C3x
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C3x
|
||||
<br><br>
|
||||
|
||||
# RepC3
|
||||
---
|
||||
:::ultralytics.nn.modules.block.RepC3
|
||||
<br><br>
|
||||
|
||||
# C3TR
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C3TR
|
||||
<br><br>
|
||||
|
||||
# C3Ghost
|
||||
---
|
||||
:::ultralytics.nn.modules.block.C3Ghost
|
||||
<br><br>
|
||||
|
||||
# GhostBottleneck
|
||||
---
|
||||
:::ultralytics.nn.modules.block.GhostBottleneck
|
||||
<br><br>
|
||||
|
||||
# Bottleneck
|
||||
---
|
||||
:::ultralytics.nn.modules.block.Bottleneck
|
||||
<br><br>
|
||||
|
||||
# BottleneckCSP
|
||||
---
|
||||
:::ultralytics.nn.modules.block.BottleneckCSP
|
||||
<br><br>
|
||||
64
docs/reference/nn/modules/conv.md
Normal file
64
docs/reference/nn/modules/conv.md
Normal file
@ -0,0 +1,64 @@
|
||||
# Conv
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.Conv
|
||||
<br><br>
|
||||
|
||||
# LightConv
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.LightConv
|
||||
<br><br>
|
||||
|
||||
# DWConv
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.DWConv
|
||||
<br><br>
|
||||
|
||||
# DWConvTranspose2d
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.DWConvTranspose2d
|
||||
<br><br>
|
||||
|
||||
# ConvTranspose
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.ConvTranspose
|
||||
<br><br>
|
||||
|
||||
# Focus
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.Focus
|
||||
<br><br>
|
||||
|
||||
# GhostConv
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.GhostConv
|
||||
<br><br>
|
||||
|
||||
# RepConv
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.RepConv
|
||||
<br><br>
|
||||
|
||||
# ChannelAttention
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.ChannelAttention
|
||||
<br><br>
|
||||
|
||||
# SpatialAttention
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.SpatialAttention
|
||||
<br><br>
|
||||
|
||||
# CBAM
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.CBAM
|
||||
<br><br>
|
||||
|
||||
# Concat
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.Concat
|
||||
<br><br>
|
||||
|
||||
# autopad
|
||||
---
|
||||
:::ultralytics.nn.modules.conv.autopad
|
||||
<br><br>
|
||||
24
docs/reference/nn/modules/head.md
Normal file
24
docs/reference/nn/modules/head.md
Normal file
@ -0,0 +1,24 @@
|
||||
# Detect
|
||||
---
|
||||
:::ultralytics.nn.modules.head.Detect
|
||||
<br><br>
|
||||
|
||||
# Segment
|
||||
---
|
||||
:::ultralytics.nn.modules.head.Segment
|
||||
<br><br>
|
||||
|
||||
# Pose
|
||||
---
|
||||
:::ultralytics.nn.modules.head.Pose
|
||||
<br><br>
|
||||
|
||||
# Classify
|
||||
---
|
||||
:::ultralytics.nn.modules.head.Classify
|
||||
<br><br>
|
||||
|
||||
# RTDETRDecoder
|
||||
---
|
||||
:::ultralytics.nn.modules.head.RTDETRDecoder
|
||||
<br><br>
|
||||
49
docs/reference/nn/modules/transformer.md
Normal file
49
docs/reference/nn/modules/transformer.md
Normal file
@ -0,0 +1,49 @@
|
||||
# TransformerEncoderLayer
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.TransformerEncoderLayer
|
||||
<br><br>
|
||||
|
||||
# AIFI
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.AIFI
|
||||
<br><br>
|
||||
|
||||
# TransformerLayer
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.TransformerLayer
|
||||
<br><br>
|
||||
|
||||
# TransformerBlock
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.TransformerBlock
|
||||
<br><br>
|
||||
|
||||
# MLPBlock
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.MLPBlock
|
||||
<br><br>
|
||||
|
||||
# MLP
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.MLP
|
||||
<br><br>
|
||||
|
||||
# LayerNorm2d
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.LayerNorm2d
|
||||
<br><br>
|
||||
|
||||
# MSDeformAttn
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.MSDeformAttn
|
||||
<br><br>
|
||||
|
||||
# DeformableTransformerDecoderLayer
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.DeformableTransformerDecoderLayer
|
||||
<br><br>
|
||||
|
||||
# DeformableTransformerDecoder
|
||||
---
|
||||
:::ultralytics.nn.modules.transformer.DeformableTransformerDecoder
|
||||
<br><br>
|
||||
24
docs/reference/nn/modules/utils.md
Normal file
24
docs/reference/nn/modules/utils.md
Normal file
@ -0,0 +1,24 @@
|
||||
# _get_clones
|
||||
---
|
||||
:::ultralytics.nn.modules.utils._get_clones
|
||||
<br><br>
|
||||
|
||||
# bias_init_with_prob
|
||||
---
|
||||
:::ultralytics.nn.modules.utils.bias_init_with_prob
|
||||
<br><br>
|
||||
|
||||
# linear_init_
|
||||
---
|
||||
:::ultralytics.nn.modules.utils.linear_init_
|
||||
<br><br>
|
||||
|
||||
# inverse_sigmoid
|
||||
---
|
||||
:::ultralytics.nn.modules.utils.inverse_sigmoid
|
||||
<br><br>
|
||||
|
||||
# multi_scale_deformable_attn_pytorch
|
||||
---
|
||||
:::ultralytics.nn.modules.utils.multi_scale_deformable_attn_pytorch
|
||||
<br><br>
|
||||
@ -27,6 +27,11 @@ description: Learn how to work with Ultralytics YOLO Detection, Segmentation & C
|
||||
:::ultralytics.nn.tasks.ClassificationModel
|
||||
<br><br>
|
||||
|
||||
# Ensemble
|
||||
---
|
||||
:::ultralytics.nn.tasks.Ensemble
|
||||
<br><br>
|
||||
|
||||
# torch_safe_load
|
||||
---
|
||||
:::ultralytics.nn.tasks.torch_safe_load
|
||||
|
||||
Reference in New Issue
Block a user