์ฐธ์กฐ ultralytics/utils/autobatch.py
์ฐธ๊ณ
์ด ํ์ผ์ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/utils/autobatch .py์์ ํ์ธํ ์ ์์ต๋๋ค. ๋ฌธ์ ๋ฅผ ๋ฐ๊ฒฌํ๋ฉด ํ ๋ฆฌํ์คํธ (๐ ๏ธ)๋ฅผ ์ ์ถํ์ฌ ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋๋ก ๋์์ฃผ์ธ์. ๊ฐ์ฌํฉ๋๋ค ๐!
ultralytics.utils.autobatch.check_train_batch_size(model, imgsz=640, amp=True)
์๋ ๋ฐฐ์น() ํจ์๋ฅผ ์ฌ์ฉํ์ฌ YOLO ๊ต์ก ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ํ์ธํฉ๋๋ค.
๋งค๊ฐ๋ณ์:
์ด๋ฆ | ์ ํ | ์ค๋ช | ๊ธฐ๋ณธ๊ฐ |
---|---|---|---|
model |
Module
|
YOLO ๋ชจ๋ธ์ ์ฌ์ฉํ์ฌ ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ํ์ธํฉ๋๋ค. |
ํ์ |
imgsz |
int
|
๊ต์ก์ ์ฌ์ฉ๋๋ ์ด๋ฏธ์ง ํฌ๊ธฐ์ ๋๋ค. |
640
|
amp |
bool
|
True์ด๋ฉด ๊ต์ก์ ์๋ ํผํฉ ์ ๋ฐ๋(AMP)๋ฅผ ์ฌ์ฉํฉ๋๋ค. |
True
|
๋ฐํํฉ๋๋ค:
์ ํ | ์ค๋ช |
---|---|
int
|
์๋ ๋ฐฐ์น() ํจ์๋ฅผ ์ฌ์ฉํ์ฌ ๊ณ์ฐ๋ ์ต์ ์ ๋ฐฐ์น ํฌ๊ธฐ์ ๋๋ค. |
์ ์์ค ์ฝ๋ ultralytics/utils/autobatch.py
ultralytics.utils.autobatch.autobatch(model, imgsz=640, fraction=0.6, batch_size=DEFAULT_CFG.batch)
์ฌ์ฉ ๊ฐ๋ฅํ CUDA ๋ฉ๋ชจ๋ฆฌ์ ์ผ๋ถ๋ฅผ ์ฌ์ฉํ๋ ์ต์ ์ YOLO ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ์๋์ผ๋ก ์ถ์ ํฉ๋๋ค.
๋งค๊ฐ๋ณ์:
์ด๋ฆ | ์ ํ | ์ค๋ช | ๊ธฐ๋ณธ๊ฐ |
---|---|---|---|
model |
module
|
YOLO ๋ชจ๋ธ์ ์ฌ์ฉํ์ฌ ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ ๊ณ์ฐํฉ๋๋ค. |
ํ์ |
imgsz |
int
|
YOLO ๋ชจ๋ธ์ ์ ๋ ฅ์ผ๋ก ์ฌ์ฉ๋๋ ์ด๋ฏธ์ง ํฌ๊ธฐ์ ๋๋ค. ๊ธฐ๋ณธ๊ฐ์ 640์ ๋๋ค. |
640
|
fraction |
float
|
์ฌ์ฉํ ์ ์๋ CUDA ๋ฉ๋ชจ๋ฆฌ์ ๋น์จ์ ๋๋ค. ๊ธฐ๋ณธ๊ฐ์ 0.60์ ๋๋ค. |
0.6
|
batch_size |
int
|
์ค๋ฅ๊ฐ ๊ฐ์ง๋ ๊ฒฝ์ฐ ์ฌ์ฉํ ๊ธฐ๋ณธ ๋ฐฐ์น ํฌ๊ธฐ์ ๋๋ค. ๊ธฐ๋ณธ๊ฐ์ 16์ ๋๋ค. |
batch
|
๋ฐํํฉ๋๋ค:
์ ํ | ์ค๋ช |
---|---|
int
|
์ต์ ์ ๋ฐฐ์น ํฌ๊ธฐ์ ๋๋ค. |