Skip to content

Instantly share code, notes, and snippets.

@gushart
gushart / convert_weights.py
Last active June 2, 2019 13:15 — forked from hollance/convert_weights.py
SE-ResNet-50 in Keras
# Convert SE-ResNet-50 from Caffe to Keras
# Using the model from https://github.com/shicai/SENet-Caffe
import os
import numpy as np
# The caffe module needs to be on the Python path; we'll add it here explicitly.
import sys
caffe_root = "/path/to/caffe"
sys.path.insert(0, caffe_root + "python")