generate.py 文件源码

def __create_network_prototxt(self, deploy):
        n = caffe.NetSpec()
        top = self.__create_data_layer(n, deploy=deploy)
        top = self.__create_first_convolutional_layer(n, top)

        # Create Residual Units
        for i in range(len(self.__num_residual_units)):
            stride = 1 if i == 0 else 2
            top = self.__layer(n, top, 'res%d' % (2+i), self.__wide_basic, self.__num_feature_maps[i],
                               self.__num_feature_maps[i+1], self.__num_residual_units[i], stride, deploy)

        top = self.__create_fully_connected_layer(n, top, deploy=deploy)
        self.__create_output_layer(n, top, deploy=deploy)

        # Return prototxt
        return str(n.to_proto()).replace('__test', '')