Baidu рдХрд╛ RT-DETR: рдПрдХ рд╡рд┐рдЬрди рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░-рдЖрдзрд╛рд░рд┐рдд рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрдЯрд░
рд╡рд┐рд╣рдВрдЧрд╛рд╡рд▓реЛрдХрди
рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдбрд┐рдЯреЗрдХреНрд╢рди рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░ (RT-DETR), Baidu рджреНрд╡рд╛рд░рд╛ рд╡рд┐рдХрд╕рд┐рдд, рдПрдХ рдЕрддреНрдпрд╛рдзреБрдирд┐рдХ рдПрдВрдб-рдЯреВ-рдПрдВрдб рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрдЯрд░ рд╣реИ рдЬреЛ рдЙрдЪреНрдЪ рд╕рдЯреАрдХрддрд╛ рдмрдирд╛рдП рд░рдЦрддреЗ рд╣реБрдП рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдХрд╛ рдкреНрд░рджрд░реНрд╢рди рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИред рдпрд╣ рд╡рд┐рдЬрди рдЯреНрд░рд╛рдВрд╕рдлреЙрд░реНрдорд░ (рд╡реАрдЖрдИрдЯреА) рдХреА рд╢рдХреНрддрд┐ рдХрд╛ рд▓рд╛рдн рдЙрдард╛рддрд╛ рд╣реИ рддрд╛рдХрд┐ рдЗрдВрдЯреНрд░рд╛-рд╕реНрдХреЗрд▓ рдЗрдВрдЯрд░реИрдХреНрд╢рди рдФрд░ рдХреНрд░реЙрд╕-рд╕реНрдХреЗрд▓ рдлреНрдпреВрдЬрди рдХреЛ рдЕрд▓рдЧ рдХрд░рдХреЗ рдорд▓реНрдЯреАрд╕реНрдХреЗрд▓ рд╕реБрд╡рд┐рдзрд╛рдУрдВ рдХреЛ рдХреБрд╢рд▓рддрд╛рдкреВрд░реНрд╡рдХ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд┐рдпрд╛ рдЬрд╛ рд╕рдХреЗред RT-DETR рдЕрддреНрдпрдзрд┐рдХ рдЕрдиреБрдХреВрд▓рдиреАрдп рд╣реИ, рдмрд┐рдирд╛ рдХрд┐рд╕реА рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреЗ рд╡рд┐рднрд┐рдиреНрди рдбрд┐рдХреЛрдбрд░ рдкрд░рддреЛрдВ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдЕрдиреБрдорд╛рди рдЧрддрд┐ рдХреЗ рд▓рдЪреАрд▓реЗ рд╕рдорд╛рдпреЛрдЬрди рдХрд╛ рд╕рдорд░реНрдерди рдХрд░рддрд╛ рд╣реИред рдореЙрдбрд▓ CUDA рдЬреИрд╕реЗ рддреНрд╡рд░рд┐рдд рдмреИрдХрдПрдВрдб рдкрд░ рдЙрддреНрдХреГрд╖реНрдЯрддрд╛ рдкреНрд░рд╛рдкреНрдд рдХрд░рддрд╛ рд╣реИ TensorRT, рдХрдИ рдЕрдиреНрдп рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрдЯрд░реЛрдВ рд╕реЗ рдмреЗрд╣рддрд░ рдкреНрд░рджрд░реНрд╢рди рдХрд░ рд░рд╣рд╛ рд╣реИред
рд╕рддрд░реНрдХрддрд╛: рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдбрд┐рдЯреЗрдХреНрд╢рди рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░ (RT-DETR)
Baidu рдХрд╛ рдЕрд╡рд▓реЛрдХрди RT-DETR. рд╡рд╣реА RT-DETR рдореЙрдбрд▓ рдЖрд░реНрдХрд┐рдЯреЗрдХреНрдЪрд░ рдЖрд░реЗрдЦ рдмреИрдХрдмреЛрди {S3, S4, S5} рдХреЗ рдЕрдВрддрд┐рдо рддреАрди рдЪрд░рдгреЛрдВ рдХреЛ рдПрдирдХреЛрдбрд░ рдХреЗ рдЗрдирдкреБрдЯ рдХреЗ рд░реВрдк рдореЗрдВ рджрд┐рдЦрд╛рддрд╛ рд╣реИред рдХреБрд╢рд▓ рд╣рд╛рдЗрдмреНрд░рд┐рдб рдПрдирдХреЛрдбрд░ рдЗрдВрдЯреНрд░рд╛рд╕реНрдХреЗрд▓ рдлреАрдЪрд░ рдЗрдВрдЯрд░реИрдХреНрд╢рди (рдПрдЖрдИрдПрдлрдЖрдИ) рдФрд░ рдХреНрд░реЙрд╕-рд╕реНрдХреЗрд▓ рдлреАрдЪрд░-рдлреНрдпреВрдЬрди рдореЙрдбреНрдпреВрд▓ (рд╕реАрд╕реАрдПрдлрдПрдо) рдХреЗ рдорд╛рдзреНрдпрдо рд╕реЗ рдорд▓реНрдЯреАрд╕реНрдХреЗрд▓ рдлреАрдЪрд░реНрд╕ рдХреЛ рдЗрдореЗрдЬ рдлреАрдЪрд░реНрд╕ рдХреЗ рдЕрдиреБрдХреНрд░рдо рдореЗрдВ рдмрджрд▓ рджреЗрддрд╛ рд╣реИред IoU-рдЬрд╛рдЧрд░реВрдХ рдХреНрд╡реЗрд░реА рдЪрдпрди рдХреЛ рдбрд┐рдХреЛрдбрд░ рдХреЗ рд▓рд┐рдП рдкреНрд░рд╛рд░рдВрднрд┐рдХ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреНрд╡реЗрд░реА рдХреЗ рд░реВрдк рдореЗрдВ рд╕реЗрд╡рд╛ рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдЫрд╡рд┐ рд╕реБрд╡рд┐рдзрд╛рдУрдВ рдХреА рдПрдХ рдирд┐рд╢реНрдЪрд┐рдд рд╕рдВрдЦреНрдпрд╛ рдХрд╛ рдЪрдпрди рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдирд┐рдпреЛрдЬрд┐рдд рдХрд┐рдпрд╛ рдЬрд╛рддрд╛ рд╣реИред рдЕрдВрдд рдореЗрдВ, рд╕рд╣рд╛рдпрдХ рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА рд╕рд┐рд░ рдХреЗ рд╕рд╛рде рдбрд┐рдХреЛрдбрд░ рдкреБрдирд░рд╛рд╡реГрддреНрдд рд░реВрдк рд╕реЗ рдмрдХреНрд╕реЗ рдФрд░ рдЖрддреНрдорд╡рд┐рд╢реНрд╡рд╛рд╕ рд╕реНрдХреЛрд░ рдЙрддреНрдкрдиреНрди рдХрд░рдиреЗ рдХреЗ рд▓рд┐рдП рдСрдмреНрдЬреЗрдХреНрдЯ рдкреНрд░рд╢реНрдиреЛрдВ рдХрд╛ рдЕрдиреБрдХреВрд▓рди рдХрд░рддрд╛ рд╣реИ (рдореВрд▓).
рдкреНрд░рдореБрдЦ рд╡рд┐рд╢реЗрд╖рддрд╛рдРрдВ
- рдХреБрд╢рд▓ рд╣рд╛рдЗрдмреНрд░рд┐рдб рдПрдирдХреЛрдбрд░: Baidu рдХрд╛ RT-DETR рдПрдХ рдХреБрд╢рд▓ рд╣рд╛рдЗрдмреНрд░рд┐рдб рдПрдирдХреЛрдбрд░ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддрд╛ рд╣реИ рдЬреЛ рдЗрдВрдЯреНрд░рд╛-рд╕реНрдХреЗрд▓ рдЗрдВрдЯрд░реИрдХреНрд╢рди рдФрд░ рдХреНрд░реЙрд╕-рд╕реНрдХреЗрд▓ рдлреНрдпреВрдЬрди рдХреЛ рдбрд┐рдХреЙрдкреНрд▓рд┐рдВрдЧ рдХрд░рдХреЗ рдорд▓реНрдЯреАрд╕реНрдХреЗрд▓ рд╕реБрд╡рд┐рдзрд╛рдУрдВ рдХреЛ рд╕рдВрд╕рд╛рдзрд┐рдд рдХрд░рддрд╛ рд╣реИред рдпрд╣ рдЕрджреНрд╡рд┐рддреАрдп рд╡рд┐рдЬрди рдЯреНрд░рд╛рдВрд╕рдлреЙрд░реНрдорд░-рдЖрдзрд╛рд░рд┐рдд рдбрд┐рдЬрд╝рд╛рдЗрди рдХрдореНрдкреНрдпреВрдЯреЗрд╢рдирд▓ рд▓рд╛рдЧрдд рдХреЛ рдХрдо рдХрд░рддрд╛ рд╣реИ рдФрд░ рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрд╢рди рдХреА рдЕрдиреБрдорддрд┐ рджреЗрддрд╛ рд╣реИред
- IoU-рдЬрд╛рдЧрд░реВрдХ рдХреНрд╡реЗрд░реА рдЪрдпрди: Baidu рдХрд╛ RT-DETR IoU-рдЬрд╛рдЧрд░реВрдХ рдХреНрд╡реЗрд░реА рдЪрдпрди рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдСрдмреНрдЬреЗрдХреНрдЯ рдХреНрд╡реЗрд░реА рдЖрд░рдВрднреАрдХрд░рдг рдореЗрдВ рд╕реБрдзрд╛рд░ рдХрд░рддрд╛ рд╣реИред рдпрд╣ рдореЙрдбрд▓ рдХреЛ рджреГрд╢реНрдп рдореЗрдВ рд╕рдмрд╕реЗ рдкреНрд░рд╛рд╕рдВрдЧрд┐рдХ рд╡рд╕реНрддреБрдУрдВ рдкрд░ рдзреНрдпрд╛рди рдХреЗрдВрджреНрд░рд┐рдд рдХрд░рдиреЗ рдХреА рдЕрдиреБрдорддрд┐ рджреЗрддрд╛ рд╣реИ, рдЬрд┐рд╕рд╕реЗ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреА рд╕рдЯреАрдХрддрд╛ рдмрдврд╝ рдЬрд╛рддреА рд╣реИред
- рдЕрдиреБрдХреВрд▓рдиреАрдп рдЕрдиреБрдорд╛рди рдЧрддрд┐: Baidu рдХрд╛ RT-DETR рдкреБрдирд░реНрдкреНрд░рд╢рд┐рдХреНрд╖рдг рдХреА рдЖрд╡рд╢реНрдпрдХрддрд╛ рдХреЗ рдмрд┐рдирд╛ рд╡рд┐рднрд┐рдиреНрди рдбрд┐рдХреЛрдбрд░ рдкрд░рддреЛрдВ рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рдХреЗ рдЕрдиреБрдорд╛рди рдЧрддрд┐ рдХреЗ рд▓рдЪреАрд▓реЗ рд╕рдорд╛рдпреЛрдЬрди рдХрд╛ рд╕рдорд░реНрдерди рдХрд░рддрд╛ рд╣реИред рдпрд╣ рдЕрдиреБрдХреВрд▓рдирд╢реАрд▓рддрд╛ рд╡рд┐рднрд┐рдиреНрди рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ рдХреЗ рдкрд░рд┐рджреГрд╢реНрдпреЛрдВ рдореЗрдВ рд╡реНрдпрд╛рд╡рд╣рд╛рд░рд┐рдХ рдЕрдиреБрдкреНрд░рдпреЛрдЧ рдХреА рд╕реБрд╡рд┐рдзрд╛ рдкреНрд░рджрд╛рди рдХрд░рддреА рд╣реИред
рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдореЙрдбрд▓
рд╡рд╣реА Ultralytics Python рдПрдкреАрдЖрдИ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ PaddlePaddle RT-DETR рд╡рд┐рднрд┐рдиреНрди рддрд░рд╛рдЬреВ рд╡рд╛рд▓реЗ рдореЙрдбрд▓:
- RT-DETR-рдПрд▓: рдХреЛрдХреЛ рд╡реИрд▓2017 рдкрд░ 53.0% рдПрдкреА, рдЯреА4 рдЬреАрдкреАрдпреВ рдкрд░ 114 рдПрдлрдкреАрдПрд╕
- RT-DETR-X: COCO val2017 рдкрд░ 54.8% AP, T4 GPU рдкрд░ 74 FPS
рдЙрдкрдпреЛрдЧ рдХреЗ рдЙрджрд╛рд╣рд░рдг
рдпрд╣ рдЙрджрд╛рд╣рд░рдг рд╕рд░рд▓ рдкреНрд░рджрд╛рди рдХрд░рддрд╛ рд╣реИ RT-DETR рдкреНрд░рд╢рд┐рдХреНрд╖рдг рдФрд░ рдЕрдиреБрдорд╛рди рдЙрджрд╛рд╣рд░рдгред рдЗрди рдФрд░ рдЕрдиреНрдп рдореЛрдб рдкрд░ рдкреВрд░реНрдг рдкреНрд░рд▓реЗрдЦрди рдХреЗ рд▓рд┐рдП рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА, рдЯреНрд░реЗрди, рд╡реИрд▓ рдФрд░ рдирд┐рд░реНрдпрд╛рдд рдбреЙрдХреНрд╕ рдкреГрд╖реНрда рджреЗрдЦреЗрдВред
рдЙрджрд╛рд╣рд░рдг
from ultralytics import RTDETR
# Load a COCO-pretrained RT-DETR-l model
model = RTDETR('rtdetr-l.pt')
# Display model information (optional)
model.info()
# Train the model on the COCO8 example dataset for 100 epochs
results = model.train(data='coco8.yaml', epochs=100, imgsz=640)
# Run inference with the RT-DETR-l model on the 'bus.jpg' image
results = model('path/to/bus.jpg')
рд╕рдорд░реНрдерд┐рдд рдХрд╛рд░реНрдп рдФрд░ рдореЛрдб
рдпрд╣ рддрд╛рд▓рд┐рдХрд╛ рдореЙрдбрд▓ рдкреНрд░рдХрд╛рд░, рд╡рд┐рд╢рд┐рд╖реНрдЯ рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рднрд╛рд░, рдкреНрд░рддреНрдпреЗрдХ рдореЙрдбрд▓ рджреНрд╡рд╛рд░рд╛ рд╕рдорд░реНрдерд┐рдд рдХрд╛рд░реНрдпреЛрдВ рдФрд░ рд╡рд┐рднрд┐рдиреНрди рдореЛрдб (рдЯреНрд░реЗрди , рд╡реИрд▓, рднрд╡рд┐рд╖реНрдпрд╡рд╛рдгреА, рдирд┐рд░реНрдпрд╛рдд) рдкреНрд░рд╕реНрддреБрдд рдХрд░рддреА рд╣реИ рдЬреЛ рд╕рдорд░реНрдерд┐рдд рд╣реИрдВ, рдЬреЛ рдЗрдореЛрдЬреА рджреНрд╡рд╛рд░рд╛ тЬЕ рдЗрдВрдЧрд┐рдд рдХрд┐рдП рдЧрдП рд╣реИрдВред
рдореЙрдбрд▓ рдкреНрд░рдХрд╛рд░ | рдкреВрд░реНрд╡-рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд рд╡рдЬрди | рд╕рдорд░реНрдерд┐рдд рдХрд╛рд░реНрдп | рдЕрдиреБрдорд╛рди | рдорд╛рдиреНрдпрддрд╛ | рдкреНрд░рд╢рд┐рдХреНрд╖рдг | рдирд┐рд░реНрдпрд╛рддрд┐рдд рдорд╛рд▓ |
---|---|---|---|---|---|---|
RT-DETR рдмреЬрд╛ | rtdetr-l.pt | рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ | тЬЕ | тЬЕ | тЬЕ | тЬЕ |
RT-DETR рдЕрддрд┐рд░рд┐рдХреНрдд-рдмрдбрд╝реЗ | rtdetr-x.pt | рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдирд╛ | тЬЕ | тЬЕ | тЬЕ | тЬЕ |
рдкреНрд░рд╢рдВрд╕рд╛ рдкрддреНрд░ рдФрд░ рдкрд╛рд╡рддреА
рдпрджрд┐ рдЖрдк Baidu рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░рддреЗ рд╣реИрдВ RT-DETR рдЕрдкрдиреЗ рд╢реЛрдз рдпрд╛ рд╡рд┐рдХрд╛рд╕ рдХрд╛рд░реНрдп рдореЗрдВ, рдХреГрдкрдпрд╛ рдореВрд▓ рдкреЗрдкрд░ рдХрд╛ рд╣рд╡рд╛рд▓рд╛ рджреЗрдВ:
рд╣рдо Baidu рдФрд░ рдХреЛ рд╕реНрд╡реАрдХрд╛рд░ рдХрд░рдирд╛ рдЪрд╛рд╣реЗрдВрдЧреЗ PaddlePaddle рдХрдВрдкреНрдпреВрдЯрд░ рджреГрд╖реНрдЯрд┐ рд╕рдореБрджрд╛рдп рдХреЗ рд▓рд┐рдП рдЗрд╕ рдореВрд▓реНрдпрд╡рд╛рди рд╕рдВрд╕рд╛рдзрди рдХреЛ рдмрдирд╛рдиреЗ рдФрд░ рдмрдирд╛рдП рд░рдЦрдиреЗ рдХреЗ рд▓рд┐рдП рдЯреАрдоред рд╡рд┐рдЬрди рдЯреНрд░рд╛рдВрд╕рдлреЙрд░реНрдорд░-рдЖрдзрд╛рд░рд┐рдд рд░реАрдпрд▓-рдЯрд╛рдЗрдо рдСрдмреНрдЬреЗрдХреНрдЯ рдбрд┐рдЯреЗрдХреНрдЯрд░ рдХреЗ рд╡рд┐рдХрд╛рд╕ рдХреЗ рд╕рд╛рде рдХреНрд╖реЗрддреНрд░ рдореЗрдВ рдЙрдирдХрд╛ рдпреЛрдЧрджрд╛рди, RT-DETR, рдмрд╣реБрдд рд╕рд░рд╛рд╣рдирд╛ рдХреА рд╣реИред
рдЦреЛрдЬрд╢рдмреНрджреЛрдВ: RT-DETR, рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░, ViT, рд╡рд┐рдЬрди рдЯреНрд░рд╛рдВрд╕рдлреЙрд░реНрдорд░, Baidu RT-DETR, PaddlePaddle, рдкреИрдбрд▓ рдкреИрдбрд▓ RT-DETR, рд╡рд╛рд╕реНрддрд╡рд┐рдХ рд╕рдордп рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ, рд╡рд┐рдЬрди рдЯреНрд░рд╛рдВрд╕рдлрд╛рд░реНрдорд░ рдЖрдзрд╛рд░рд┐рдд рд╡рд╕реНрддреБ рдХрд╛ рдкрддрд╛ рд▓рдЧрд╛рдиреЗ, рдкреВрд░реНрд╡ рдкреНрд░рд╢рд┐рдХреНрд╖рд┐рдд PaddlePaddle RT-DETR рдореЙрдбрд▓, Baidu's RT-DETR рдЙрдкрдпреЛрдЧ Ultralytics Python рдПрдкреАрдЖрдИ
2023-11-12 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-04-17
рд▓реЗрдЦрдХ: рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (9), рд░рд┐рдЬрд╡рд╛рдирдореБрдирд╡реНрд╡рд░ (1)