pytorchTocaffe

本代码实现pytorch模型转换为caffe模型

generate_prototxt.py

pytorchToCaffe.py

test_ponet.py

test_ponet_caffe.py

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
12		12
24		24
__pycache__		__pycache__
data_process		data_process
layer		layer
lmark		lmark
models		models
network		network
test		test
util_tf		util_tf
utils		utils
MTCNN_nets.py		MTCNN_nets.py
README.md		README.md
extract_caffe_blobs.py		extract_caffe_blobs.py
extract_caffe_weights.py		extract_caffe_weights.py
generate_prototxt.py		generate_prototxt.py
generate_prototxt_fpn.py		generate_prototxt_fpn.py
landmark.caffemodel		landmark.caffemodel
landmark.prototxt		landmark.prototxt
mobielnetv2_final_fpn.pth		mobielnetv2_final_fpn.pth
mobilenetv2_final_fpn.caffemodel		mobilenetv2_final_fpn.caffemodel
mobilenetv2_fpn.py		mobilenetv2_fpn.py
onet.caffemodel		onet.caffemodel
onet.prototxt		onet.prototxt
pnet.caffemodel		pnet.caffemodel
pnet.prototxt		pnet.prototxt
pytorchToCaffe.py		pytorchToCaffe.py
pytorchToCaffe_fpn.py		pytorchToCaffe_fpn.py
ssd_mobilenetv2_fpn.py		ssd_mobilenetv2_fpn.py
ssd_mobilenetv2_fpn_full.prototxt		ssd_mobilenetv2_fpn_full.prototxt
ssd_mobilenetv2_fpn_ok.prototxt		ssd_mobilenetv2_fpn_ok.prototxt
test_ponet.py		test_ponet.py
test_ponet_caffe.py		test_ponet_caffe.py