diff --git a/Caffe/layer_param.py b/Caffe/layer_param.py index fd966d7..5cbe56c 100755 --- a/Caffe/layer_param.py +++ b/Caffe/layer_param.py @@ -26,7 +26,12 @@ def __init__(self,name='',type='',top=(),bottom=()): self.param=pb.LayerParameter() self.name=self.param.name=name self.type=self.param.type=type - + + if None in top: + top = list(filter(None, top)) + if None in bottom: + bottom = list(filter(None, bottom)) + self.top=self.param.top self.top.extend(top) self.bottom=self.param.bottom @@ -180,4 +185,4 @@ def copy_from(self,layer_param): pass def set_enum(param,key,value): - setattr(param,key,param.Value(value)) \ No newline at end of file + setattr(param,key,param.Value(value))