์ฐธ์กฐ ultralytics/models/rtdetr/model.py
์ฐธ๊ณ
์ด ํ์ผ์ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py์์ ํ์ธํ ์ ์์ต๋๋ค. ๋ฌธ์ ๋ฅผ ๋ฐ๊ฒฌํ๋ฉด ํ ๋ฆฌํ์คํธ (๐ ๏ธ) ๋ฅผ ํตํด ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋๋ก ๋์์ฃผ์ธ์. ๊ฐ์ฌํฉ๋๋ค ๐!
ultralytics.models.rtdetr.model.RTDETR
๋ฒ ์ด์ค: Model
๋ฐ์ด๋์ RT-DETR ๋ชจ๋ธ์ฉ ์ธํฐํ์ด์ค์ ๋๋ค. ์ด ๋น์ ํธ๋์คํฌ๋จธ ๊ธฐ๋ฐ ๋ฌผ์ฒด ๊ฐ์ง๊ธฐ๋ ๋์ ์ ํ๋๋ก ์ค์๊ฐ ์ฑ๋ฅ์ ์ ๊ณตํฉ๋๋ค. ์ค์๊ฐ ์ฑ๋ฅ์ ์ ๊ณตํฉ๋๋ค. ํจ์จ์ ์ธ ํ์ด๋ธ๋ฆฌ๋ ์ธ์ฝ๋ฉ, IoU ์ธ์ ์ฟผ๋ฆฌ ์ ํ, ์ ์ํ ์ถ๋ก ์๋๋ฅผ ์ง์ํฉ๋๋ค.
์์ฑ:
์ด๋ฆ | ์ ํ | ์ค๋ช |
---|---|---|
model |
str
|
์ฌ์ ํ์ต๋ ๋ชจ๋ธ์ ๊ฒฝ๋ก์ ๋๋ค. ๊ธฐ๋ณธ๊ฐ์ 'rtdetr-l.pt'์ ๋๋ค. |
์ ์์ค ์ฝ๋ ultralytics/models/rtdetr/model.py
task_map: dict
property
RT-DETR ์ ๋ํ ์์ ๋งต์ ๋ฐํํ์ฌ ์์ ์ ํด๋น Ultralytics ํด๋์ค์ ์ฐ๊ฒฐํฉ๋๋ค.
๋ฐํํฉ๋๋ค:
์ด๋ฆ | ์ ํ | ์ค๋ช |
---|---|---|
dict |
dict
|
RT-DETR ๋ชจ๋ธ์ ๋ํด ์์ ์ด๋ฆ์ Ultralytics ์์ ํด๋์ค์ ๋งคํํ๋ ์ฌ์ ์ ๋๋ค. |
__init__(model='rtdetr-l.pt')
์ง์ ๋ ์ฌ์ ํ์ต๋ ๋ชจ๋ธ ํ์ผ๋ก RT-DETR ๋ชจ๋ธ์ ์ด๊ธฐํํฉ๋๋ค. .pt ๋ฐ .yaml ํ์์ ์ง์ํฉ๋๋ค.
๋งค๊ฐ๋ณ์:
์ด๋ฆ | ์ ํ | ์ค๋ช | ๊ธฐ๋ณธ๊ฐ |
---|---|---|---|
model |
str
|
์ฌ์ ํ์ต๋ ๋ชจ๋ธ์ ๊ฒฝ๋ก์ ๋๋ค. ๊ธฐ๋ณธ๊ฐ์ 'rtdetr-l.pt'์ ๋๋ค. |
'rtdetr-l.pt'
|
์ฌ๋ฆฌ๋ค:
์ ํ | ์ค๋ช |
---|---|
NotImplementedError
|
๋ชจ๋ธ ํ์ผ ํ์ฅ์๊ฐ 'pt', 'yaml' ๋๋ 'yml'์ด ์๋ ๊ฒฝ์ฐ. |