使用keras已经实现的神经网络时只能进行fine_tune,有的时候需要对网络结构进行修改,或者在keras中添加自己的网络结构。
过程如下:以InceptionV3为例
1.将修改好的网络结构,例如keras_applications/Inception_v3_s.py放入keras_applications目录中
2.在keras/applications/Inception_v3.py中添加导入代码
from keras_applications import inception_v3_s
继续添加模块注册代码
@keras_modules_injection
def InceptionV3s(*args, **kwargs):
return inception_v3_s.InceptionV3s(*args, **kwargs)
3.在keras/applications/__init__.py中添加注册代码
from .inception_v3 import InceptionV3s