์ฐธ์กฐ ultralytics/models/yolo/classify/predict.py
์ฐธ๊ณ
์ด ํ์ผ์ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ yolo/classify/predict .py์์ ํ์ธํ ์ ์์ต๋๋ค. ๋ฌธ์ ๋ฅผ ๋ฐ๊ฒฌํ๋ฉด ํ ๋ฆฌํ์คํธ (๐ ๏ธ) ๋ฅผ ํตํด ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋๋ก ๋์์ฃผ์ธ์. ๊ฐ์ฌํฉ๋๋ค ๐!
ultralytics.models.yolo.classify.predict.ClassificationPredictor
๋ฒ ์ด์ค: BasePredictor
๋ถ๋ฅ ๋ชจ๋ธ์ ๊ธฐ๋ฐํ ์์ธก์ ์ํด BasePredictor ํด๋์ค๋ฅผ ํ์ฅํ ํด๋์ค์ ๋๋ค.
์ฐธ๊ณ
- ํ ์น๋น์ ๋ถ๋ฅ ๋ชจ๋ธ์ 'model' ์ธ์์ ์ ๋ฌํ ์๋ ์์ต๋๋ค(์: model='resnet18').
์
์ ์์ค ์ฝ๋ ultralytics/models/yolo/classify/predict.py
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
์์ ์ '๋ถ๋ฅ'๋ก ์ค์ ํ ClassificationPredictor๋ฅผ ์ด๊ธฐํํฉ๋๋ค.
์ ์์ค ์ฝ๋ ultralytics/models/yolo/classify/predict.py
postprocess(preds, img, orig_imgs)
์์ธก์ ํ์ฒ๋ฆฌํ์ฌ ๊ฒฐ๊ณผ ๊ฐ์ฒด๋ฅผ ๋ฐํํฉ๋๋ค.
์ ์์ค ์ฝ๋ ultralytics/models/yolo/classify/predict.py
preprocess(img)
์ ๋ ฅ ์ด๋ฏธ์ง๋ฅผ ๋ชจ๋ธ ํธํ ๋ฐ์ดํฐ ์ ํ์ผ๋ก ๋ณํํฉ๋๋ค.