Caffe Documentation : Caffe Model Zoo (Kor) - ys7yoo/BrainCaffe GitHub Wiki

Caffe ๋ชจ๋ธ Zoo (Caffe Model Zoo)

๋งŽ์€ ์—ฐ๊ตฌ์ž๋“ค๊ณผ ๊ณตํ•™์ž๋“ค์€ ๋ชจ๋“  ์ข…๋ฅ˜์˜ ๊ตฌ์กฐ์™€ ๋ฐ์ดํ„ฐ๋ฅผ ์‚ฌ์šฉํ•ด์„œ ๋‹ค์–‘ ์—…๋ฌด์— ๋Œ€ํ•œ Caffe ๋ชจ๋ธ๋“ค์„ ์ œ์ž‘ํ•ด์™”๋‹ค. ์ด๋Ÿฌํ•œ ๋ชจ๋ธ๋“ค์€ ๋‹จ์ˆœ ํšŒ๊ท€๋ถ€ํ„ฐ ํฐ ํฌ๊ธฐ์˜ ์‹œ๊ฐ ๋ถ„๋ฅ˜, ์ด๋ฏธ์ง€ ์œ ์‚ฌ๋„์— ๋Œ€ํ•œ ์ƒด ๋„คํŠธ์›Œํฌ, ์—ฐ์„ค๊ณผ ๋กœ๋ด‡๊ณตํ•™์‘์šฉ๊นŒ์ง€์˜ ๋ฌธ์ œ ๋ฒ”์œ„์— ์ ์šฉํ•˜๊ณ  ํ•™์Šตํ–ˆ๋‹ค.

์ด๋Ÿฌํ•œ ๋ชจ๋ธ์„ ๊ณต์œ ํ•˜๋Š”๋ฐ ๋„์›€์„ ์ฃผ๊ธฐ์œ„ํ•ด, ์šฐ๋ฆฌ๋Š” ๋ชจ๋ธ Zoo ํ”„๋ ˆ์ž„ ์›Œํฌ๋ฅผ ์†Œ๊ฐœํ•˜๊ณ ์žํ•œ๋‹ค.

  • Caffe ๋ชจ๋ธ ์ •๋ณด๋ฅผ ๋‹ด๊ณ ์žˆ๋Š” ํ‘œ์ค€ ํ˜•ํƒœ
  • Github Gists ์—/๋กœ๋ถ€ํ„ฐ ๋ชจ๋ธ์ •๋ณด๋ฅผ ์—…๋กœ๋“œ/๋‹ค์šด๋กœ๋“œ ํ•˜๋Š” ํˆด
  • Gists์— ๋ชจ๋ธ ์ •๋ณด๋ฅผ ๊ณต์œ ํ•˜๊ธฐ์œ„ํ•œ ์ค‘์‹ฌ ์œ„ํ‚ค ํŽ˜์ด์ง€

์–ด๋””์„œ ํ•™์Šต๋œ ๋ชจ๋ธ์„ ์–ป๋Š”๊ฐ€ (Where to get trained models)

์ผ๋‹จ, ์šฐ๋ฆฌ๋Š” ์ œํ•œ์„ ๋ฐ›์ง€์•Š๊ณ  ๋ฐ•์Šค์‚ฌ์šฉ์—์„œ ๋ฒ—์–ด๋‚œ 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 ์‚ฌ์šฉ์ž๋“ค์— ์˜ํ•ด ๋งŒ๋“ค์–ด์ง„ ๊ณต๋™์˜ ๋ชจ๋ธ์€ ๋ˆ„๊ตฌ๋‚˜ ๊ณ ์น ์ˆ˜ ์žˆ๋Š” ์œ„ํ‚ค ํŽ˜์ด์ง€์— ๊ฒŒ์‹œ๋˜์–ด์žˆ๋‹ค. ์ด๋Ÿฌํ•œ ๋ชจ๋ธ์€ ์ธ์šฉ๊ตฌ์™€ ๋ผ์ด์„ผ์Šค ๊ฐ™์€ ๊ทธ๋“ค ๊ฐ๊ฐ์˜ ์ €์ž๋“ค์˜ ์ปจ๋””์…˜์— ๋Œ€์ƒ์ด๋‹ค. ๋‹น์‹ ์˜ ๋ชจ๋ธ์„ ๊ณต์œ ํ•ด ์ฃผ์–ด์„œ ๊ฐ์‚ฌํ•˜๋‹ค.

๋ชจ๋ธ ์ •๋ณด ํ˜•ํƒœ (Model info format)

Caffe ๋ชจ๋ธ์€ ๋””๋ ‰ํ† ๋ฆฌ์— ํฌํ•จ๋œ ๊ฒƒ์œผ๋กœ ์ฃผ์–ด์ง„๋‹ค.

  • ํ•ด๊ฒฐ์‚ฌ/๋ชจ๋ธ prototxt
  • readme.md ํฌํ•จ
  • YAML ์ „๋ฌธ
    • ์ด๋Ÿฐ ๋ชจ๋ธ์„ ํ•™์Šตํ•˜๊ธฐ ์œ„ํ•œ Caffe ๋ฒ„์ „ (ํ•ด์‰ฌํƒœ๊ทธ ํ˜น์€ ๋ฆด๋ฆฌ์ฆˆ ํƒœ๊ทธ)
    • [์ถ”๊ฐ€์˜ต์…˜] ํ•™์Šต๋œ .caffemodel์˜ URL์™€ SHA1ํŒŒ์ผ
    • [์ถ”๊ฐ€์˜ต์…˜] github gist ์•„์ด๋””
  • ์–ด๋–ค ๋ฐ์ดํ„ฐ์™€ ๋ชจ๋ธ์ด ๋ชจ๋ธ๋ง ์„ ํƒ ๋“ฑ์— ๋Œ€ํ•˜์—ฌ ํ•™์Šตํ–ˆ๋Š”์ง€์— ๊ด€ํ•œ ์ •๋ณด
  • ๋ผ์ด์„ผ์Šค ์ •๋ณด
  • ๋‹ค๋ฅธ ๋„์›€์  ์ •๋ณด๋“ค

๋ชจ๋ธ ์ •๋ณด ํ˜ธ์ŠคํŒ… (Hosting model info)

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> ๋กœ ์‰ฝ๊ฒŒ ๋  ๊ฒƒ์ด๋‹ค.

ํ•™์Šต๋œ ๋ชจ๋ธ ํ˜ธ์ŠคํŒ… (Hosting trained models)

.caffemodel ํŒŒ์ผ์— ์–ด๋””์— ํ˜ธ์ŠคํŠธ์‹œํ‚ฌ์ง€๋Š” ์œ ์ €ํ•œํ…Œ ๋‹ฌ๋ ค์žˆ๋‹ค. ์šฐ๋ฆฌ๋Š” BVLC์—์„œ ์ œ๊ณตํ•œ ๋ณด๋ธ์„ ์šฐ๋ฆฌ๋งŒ์˜ ์„œ๋ฒ„๋ ˆ ํ˜ธ์ŠคํŠธ ์‹œ์ผฐ๋‹ค. ๋“œ๋กญ๋ฐ•์Šค(Dropbox)๋˜ํ•œ ๊ดœ์ฐฎ๋‹ค.(ํŒ:?dl=1์ด URL ๊ธ‘์— ์ถ”๊ฐ€๋˜๋Š”๊ฒƒ์„ ํ™•์‹คํžˆ ํ•˜๋ผ.)

scripts/download_model_binary.py ๋Š” /readme.md์— ๋ช…์‹œ๋œ URL๋กœ๋ถ€ํ„ฐ .caffemodel์„ ๋‹ค์šด๋กœ๋“œํ•˜๊ณ  SHA1๋ฅผ ํ™•์ธํ•œ๋‹ค.

BVLC ๋ชจ๋ธ ๋ผ์ด์„ผ์Šค (BVLC model license)

BVLC์— ์˜ํ•ด ๋ฒˆ๋“ค๋กœ ์ œ๊ณต๋˜๋Š” Caffe ๋ชจ๋ธ์€ ์ œํ•œ์—†์ด ์‚ฌ์šฉ๊ฐ€๋Šฅํ•˜๋‹ค.

์ด๋Ÿฌํ•œ ๋ชจ๋ธ๋“ค์€ ์ด๋ฏธ์ง€๋ง ํ”„๋กœ์ ํŠธ์˜ ๋ฐ์ดํ„ฐ์™€ ์ €์ž‘๊ถŒ์˜ ๋Œ€์ƒ์ผ์ˆ˜๋„ ์žˆ๋Š” ์ธํ„ฐ๋„ท ์‚ฌ์ง„์„ ํฌํ•จํ•™ ํ•™์Šต๋ฐ์ดํ„ฐ๋กœ ํ•™์Šตํ–ˆ๋‹ค.

์šฐ๋ฆฌ์˜ ํ˜„์žฌ ์—ฐ๊ตฌ์ž๋“ค๋กœ์จ ์ดํ•ด๋Š” ์ด๋Ÿฌํ•œ ํ•™์Šต๋œ ๋ชจ๋ธ ๊ฐ€์ค‘์น˜์˜ ๊ณต๊ฐœ๋ฐฐํฌ๋กœ ๋†“์—ฌ์žˆ์–ด ์ œํ•œ์ด ์—†๋‹ค๋Š” ๊ฒƒ์ธ๋ฐ, ์ด๋Š” ์˜ค๋ฆฌ์ง€๋„ ์ด๋ฏธ์ง€๋“ค์ด ์ „์ฒด์ ์ด๋“  ๋ถ€๋ถ„์ ์ด๋“  ๋ฐฐํฌ๋˜์ง€ ์•Š๋Š”๊ฒŒ ์—†๊ธฐ ๋•Œ๋ฌธ์ด๋‹ค. ๊ฐ€์ค‘์น˜๊ฐ€ ์˜ค๋ฆฌ์ง€๋„ ์ €์ž‘๊ถŒ ์†Œ์œ ์ž์˜ ์ด์ฐจ์  ์ €์ž‘๋ฌผ์ด๊ณ  ๊ทธ๋“ค์ด ์ด๋Ÿฌํ•œ ์ €์ž‘๊ถŒ์„ ๋ชจ์€๋‹ค๋Š” ํ•ด์„์ด ์ผ์–ด๋‚จ์„ ํ™•์žฅ์‹œํ‚ค๊ธฐ ์œ„ํ•ด, UC Berkeley๋Š” ์ œํ•œ์—†์ด ๊ฐ€๋Šฅํ•œ ๋„๋ฆฌ ์ด ํˆด๊ณผ ์ง€์‹์„ ํผํŠธ๋ฆฌ๊ธฐ ์œ„ํ•œ ๋Œ€ํ•™์˜ ํ•™๋ฌธ์ ์ธ ๋ฏธ์…˜์—์„œ ๊ณต์ •ํ•œ ์‚ฌ์šฉ์ •์‹ ์œผ๋กœ ์šฐ๋ฆฌ์˜ ํ˜„์žฌ ๋ฐฐํฌ๋ฅผ ๊ณ ๋ คํ•˜๋Š”๊ฒƒ์„ ์ฑ…์ž„์ง€๊ธฐ๋ณด๋‹ค ํ˜„์žฌ ๋ฐฐํฌ๋ฅผ ์‚ฌ์šฉํ•˜๋Š” ๊ฒƒ์ด ๋‹ค๋ฅธ์ด๋“ค์—๊ฒŒ ํ—ˆ๋ฝ๋˜๋Š” ๊ฒƒ์— ๊ด€ํ•ด์„œ๋Š” ์ฑ…์ž„์„์ง€์ง€ ์•Š๋Š”๋‹ค.

โš ๏ธ **GitHub.com Fallback** โš ๏ธ