nilboy / extract-caffe-params

extract caffe model's parameters to numpy array, and write them to files

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Extract-caffe-params

By nilboy

extract caffe model's parameters to numpy array, and write them to files

python extract.py -h

usage: extract.py [-h] [--model MODEL] [--weights WEIGHTS] [--output OUTPUT]

optional arguments:

  -h, --help         show this help message and exit
  --model MODEL      model prototxt path .prototxt
  --weights WEIGHTS  caffe model weights path .caffemodel
  --output OUTPUT    output path

##Example python extract.py --model=init.prototxt --weights=init.caffemodel --output=out

About

extract caffe model's parameters to numpy array, and write them to files


Languages

Language:Python 100.0%