Caffe Documentation : Caffe Model Zoo (Kor) - ys7yoo/BrainCaffe GitHub Wiki
Caffe ๋ชจ๋ธ Zoo (Caffe Model Zoo)
๋ง์ ์ฐ๊ตฌ์๋ค๊ณผ ๊ณตํ์๋ค์ ๋ชจ๋ ์ข ๋ฅ์ ๊ตฌ์กฐ์ ๋ฐ์ดํฐ๋ฅผ ์ฌ์ฉํด์ ๋ค์ ์ ๋ฌด์ ๋ํ Caffe ๋ชจ๋ธ๋ค์ ์ ์ํด์๋ค. ์ด๋ฌํ ๋ชจ๋ธ๋ค์ ๋จ์ ํ๊ท๋ถํฐ ํฐ ํฌ๊ธฐ์ ์๊ฐ ๋ถ๋ฅ, ์ด๋ฏธ์ง ์ ์ฌ๋์ ๋ํ ์ด ๋คํธ์ํฌ, ์ฐ์ค๊ณผ ๋ก๋ด๊ณตํ์์ฉ๊น์ง์ ๋ฌธ์ ๋ฒ์์ ์ ์ฉํ๊ณ ํ์ตํ๋ค.
์ด๋ฌํ ๋ชจ๋ธ์ ๊ณต์ ํ๋๋ฐ ๋์์ ์ฃผ๊ธฐ์ํด, ์ฐ๋ฆฌ๋ ๋ชจ๋ธ Zoo ํ๋ ์ ์ํฌ๋ฅผ ์๊ฐํ๊ณ ์ํ๋ค.
- Caffe ๋ชจ๋ธ ์ ๋ณด๋ฅผ ๋ด๊ณ ์๋ ํ์ค ํํ
- Github Gists ์/๋ก๋ถํฐ ๋ชจ๋ธ์ ๋ณด๋ฅผ ์ ๋ก๋/๋ค์ด๋ก๋ ํ๋ ํด
- Gists์ ๋ชจ๋ธ ์ ๋ณด๋ฅผ ๊ณต์ ํ๊ธฐ์ํ ์ค์ฌ ์ํค ํ์ด์ง
์ผ๋จ, ์ฐ๋ฆฌ๋ ์ ํ์ ๋ฐ์ง์๊ณ ๋ฐ์ค์ฌ์ฉ์์ ๋ฒ์ด๋ BVLC์ ๋ณด๋๋ค. ์์ธํ์ฌํญ์ BVLC ๋ชจ๋ธ ๋ผ์ด์ผ์ค์ ๋ณด๋ผ ์ด๋ฐ ๊ฒ์์ ๊ฐ๊ฐ์ ๊ฒ๋ค์ ์๋์์ ๋ช ์๋ scripts/download_model_binary.py where ๋ฅผ ์คํ์์ผ์ ๋ค์ด๋ก๋ ํ ์์๋ค.
- models/bvlc_reference_caffenet์ BVLC ์ฐธ์กฐ CaffeNet
์๋ ์ค๋ง(AlexNet)์ 2012๋ NIPS์์ ํฌ๋ฆฌ์ณ ์คํค์ ๊ทธ์ ๋๋ฃ๋ค์ด ๋ง๋ ์ฌ์ธต ์ปจ๋ณผ๋ฃจ์ ์ ๊ฒฝ ๋คํธ์ํฌ๋ฅผ ์ฌ์ฉํ๋ ์ด๋ฏธ์ง๋ง ๋ถ๋ฅํ์์ ๋ฌ์ฌ๋ ๋ฒ์ ผ์ผ๋ก๋ถํฐ์ ์ฌ์ํ ์ฐจ์ด์ ์ ์ฌ์ฉํด ILSVRC 2012์ ํ์ต์์ผฐ๋ค. (Trained by Jeff Donahue @jeffdonahue)
- models/bvlc_alexnet์ BVLC ์๋ ์ค๋ง
์๋ ์ค๋ง(AlexNet)์ 2012๋ NIPS์์ ํฌ๋ฆฌ์ณ ์คํค์ ๊ทธ์ ๋๋ฃ๋ค์ด ๋ง๋ ์ฌ์ธต ์ปจ๋ณผ๋ฃจ์ ์ ๊ฒฝ ๋คํธ์ํฌ๋ฅผ ์ฌ์ฉํ๋ ์ด๋ฏธ์ง๋ง ๋ถ๋ฅํ์์ ๋ฌ์ฌ๋ ๊ฒ๊ณผ ๊ฑฐ์ ์ ํํ๊ฒ ILSVRC 2012๋ฅผ ํ์ตํ์๋ค.(Trained by Evan Shelhamer @shelhamer)
-
models/bvlc_reference_rcnn_ilsvrc13์ BVLC ์ฐธ์กฐ R-CNN ILSVRC-2013
2014๋ CVPR์์ ๊ทธ๋ฆฌ์ฌํฌ์ ๊ทธ์ ๋๋ฃ๋ค์ด ๋ฌ์ฌํ R-CNN์ Caffe ๊ตฌํ (Trained by Ross Girshick @rbgirshick)
-
models/bvlc_googlenet์ BVLC GoogLeNet GoogLeNet์ 2014๋ ILSVRC์์ ์ค์ฒด๊ฒ๋์ ๊ทธ์ ๋๋ฃ๋ค์ด ๋ง๋ ์ปจ๋ณผ๋ฃจ์ ์ ์ฌ์ฉํด ๋์ฑ ๊น์ด ๊ฐ๊ธฐ์์ ๋ฌ์ฌ๋ ๊ฒ๊ณผ ๊ฑฐ์ ์ ํํ๊ฒ ILSVRC 2012๋ฅผ ํ์ตํ์๋ค. (Trained by Sergio Guadarrama @sguada)
Caffe ์ฌ์ฉ์๋ค์ ์ํด ๋ง๋ค์ด์ง ๊ณต๋์ ๋ชจ๋ธ์ ๋๊ตฌ๋ ๊ณ ์น ์ ์๋ ์ํค ํ์ด์ง์ ๊ฒ์๋์ด์๋ค. ์ด๋ฌํ ๋ชจ๋ธ์ ์ธ์ฉ๊ตฌ์ ๋ผ์ด์ผ์ค ๊ฐ์ ๊ทธ๋ค ๊ฐ๊ฐ์ ์ ์๋ค์ ์ปจ๋์ ์ ๋์์ด๋ค. ๋น์ ์ ๋ชจ๋ธ์ ๊ณต์ ํด ์ฃผ์ด์ ๊ฐ์ฌํ๋ค.
Caffe ๋ชจ๋ธ์ ๋๋ ํ ๋ฆฌ์ ํฌํจ๋ ๊ฒ์ผ๋ก ์ฃผ์ด์ง๋ค.
- ํด๊ฒฐ์ฌ/๋ชจ๋ธ prototxt
- readme.md ํฌํจ
- YAML ์ ๋ฌธ
- ์ด๋ฐ ๋ชจ๋ธ์ ํ์ตํ๊ธฐ ์ํ Caffe ๋ฒ์ (ํด์ฌํ๊ทธ ํน์ ๋ฆด๋ฆฌ์ฆ ํ๊ทธ)
- [์ถ๊ฐ์ต์ ] ํ์ต๋ .caffemodel์ URL์ SHA1ํ์ผ
- [์ถ๊ฐ์ต์ ] github gist ์์ด๋
- ์ด๋ค ๋ฐ์ดํฐ์ ๋ชจ๋ธ์ด ๋ชจ๋ธ๋ง ์ ํ ๋ฑ์ ๋ํ์ฌ ํ์ตํ๋์ง์ ๊ดํ ์ ๋ณด
- ๋ผ์ด์ผ์ค ์ ๋ณด
- ๋ค๋ฅธ ๋์์ ์ ๋ณด๋ค
Github Gist๋ ๋ชจ๋ธ ์ ๋ณด์ ์ ํต์ง๋ก ์ข์ ํํ์ธ๋ฐ ์ด๋ ์ฌ๋ฌ ํ์ผ๋ค์ ๋ด์์ ์๊ณ , ๋ฒ์ ์์ ์ด ๊ฐ๋ฅํ๋ฉฐ, ๋ธ๋ผ์ฐ์ ์์ ์ ํ์ค(syntax)๋ฅผ ๊ฐ์กฐํ๋ฉฐ ๋ ๋๋ง์ ๊ฐ์ ์ธํ์ํจ๋ค.
scripts/upload_model_to_gist.sh ๋ Github Gist๋ก ๋ชจ๋ธ ๋๋ ํ ๋ฆฌ ์ ์ด์ง์๊ฐ ์๋ ํ์ผ์ ์ ๋ก๋ํ๋ฉฐ Gist ์์ด๋๋ฅผ ์ถ๋ ฅํ๋ค. ๋ง์ฝ gist_id๊ฐ ์ด๋ฏธ /readme.md์ ์ผ๋ถ๋ผ๋ฉด, ์กด์ฌํ๋ Gist๋ฅผ ์ ๋ฐ์ดํธ ํ๋ฉด๋๋ค.
์ ๋ก๋๋ฅผ ํ ์คํธํ๊ธฐ ์ํด์๋ scripts/upload_model_to_gist.sh models/bvlc_alexnet๋ฅผ ์๋ํด๋ณด๋ผ(ํ์ ์ ๋ก๋๋ gist๋ฅผ ์ญ์ ํ๋ ๊ฒ์ ์์ง๋ง๋ผ)
๋ชจ๋ธ์ ๋ณด๋ฅผ ๋ค์ด๋ก๋ํ๋๊ฒ์ scripts/download_model_from_gist.sh <gist_id> ๋ก ์ฝ๊ฒ ๋ ๊ฒ์ด๋ค.
.caffemodel ํ์ผ์ ์ด๋์ ํธ์คํธ์ํฌ์ง๋ ์ ์ ํํ ๋ฌ๋ ค์๋ค. ์ฐ๋ฆฌ๋ BVLC์์ ์ ๊ณตํ ๋ณด๋ธ์ ์ฐ๋ฆฌ๋ง์ ์๋ฒ๋ ํธ์คํธ ์์ผฐ๋ค. ๋๋กญ๋ฐ์ค(Dropbox)๋ํ ๊ด์ฐฎ๋ค.(ํ:?dl=1์ด URL ๊ธ์ ์ถ๊ฐ๋๋๊ฒ์ ํ์คํ ํ๋ผ.)
scripts/download_model_binary.py ๋ /readme.md์ ๋ช ์๋ URL๋ก๋ถํฐ .caffemodel์ ๋ค์ด๋ก๋ํ๊ณ SHA1๋ฅผ ํ์ธํ๋ค.
BVLC์ ์ํด ๋ฒ๋ค๋ก ์ ๊ณต๋๋ Caffe ๋ชจ๋ธ์ ์ ํ์์ด ์ฌ์ฉ๊ฐ๋ฅํ๋ค.
์ด๋ฌํ ๋ชจ๋ธ๋ค์ ์ด๋ฏธ์ง๋ง ํ๋ก์ ํธ์ ๋ฐ์ดํฐ์ ์ ์๊ถ์ ๋์์ผ์๋ ์๋ ์ธํฐ๋ท ์ฌ์ง์ ํฌํจํ ํ์ต๋ฐ์ดํฐ๋ก ํ์ตํ๋ค.
์ฐ๋ฆฌ์ ํ์ฌ ์ฐ๊ตฌ์๋ค๋ก์จ ์ดํด๋ ์ด๋ฌํ ํ์ต๋ ๋ชจ๋ธ ๊ฐ์ค์น์ ๊ณต๊ฐ๋ฐฐํฌ๋ก ๋์ฌ์์ด ์ ํ์ด ์๋ค๋ ๊ฒ์ธ๋ฐ, ์ด๋ ์ค๋ฆฌ์ง๋ ์ด๋ฏธ์ง๋ค์ด ์ ์ฒด์ ์ด๋ ๋ถ๋ถ์ ์ด๋ ๋ฐฐํฌ๋์ง ์๋๊ฒ ์๊ธฐ ๋๋ฌธ์ด๋ค. ๊ฐ์ค์น๊ฐ ์ค๋ฆฌ์ง๋ ์ ์๊ถ ์์ ์์ ์ด์ฐจ์ ์ ์๋ฌผ์ด๊ณ ๊ทธ๋ค์ด ์ด๋ฌํ ์ ์๊ถ์ ๋ชจ์๋ค๋ ํด์์ด ์ผ์ด๋จ์ ํ์ฅ์ํค๊ธฐ ์ํด, UC Berkeley๋ ์ ํ์์ด ๊ฐ๋ฅํ ๋๋ฆฌ ์ด ํด๊ณผ ์ง์์ ํผํธ๋ฆฌ๊ธฐ ์ํ ๋ํ์ ํ๋ฌธ์ ์ธ ๋ฏธ์ ์์ ๊ณต์ ํ ์ฌ์ฉ์ ์ ์ผ๋ก ์ฐ๋ฆฌ์ ํ์ฌ ๋ฐฐํฌ๋ฅผ ๊ณ ๋ คํ๋๊ฒ์ ์ฑ ์์ง๊ธฐ๋ณด๋ค ํ์ฌ ๋ฐฐํฌ๋ฅผ ์ฌ์ฉํ๋ ๊ฒ์ด ๋ค๋ฅธ์ด๋ค์๊ฒ ํ๋ฝ๋๋ ๊ฒ์ ๊ดํด์๋ ์ฑ ์์์ง์ง ์๋๋ค.