Ejemplo n.º 1
0
    def __init__(self, patch_shape, kernel, hyperparameters):
        self.cropper1d = Cropper(patch_shape[:1], kernel, hyperparameters, name="cropper1d")
        self.cropper3d = Cropper(patch_shape    , kernel, hyperparameters, name="cropper3d")
        self.patch_shape = patch_shape
        self.n_spatial_dims = len(patch_shape)

#        self.fc_conv = masonry.construct_cnn(
#            name="fc_conv",
#            layer_specs=[
#            ],
#            input_shape=(patch_shape[0], 1),
#            n_channels=4096,
#            batch_normalize=hyperparameters["batch_normalize_patch"])
        self.conv_conv = masonry.construct_cnn(
            name="fc_conv",
            layer_specs=[
                dict(size=(5, 1, 1), num_filters=512, pooling_size=(2, 1, 1), pooling_step=(2, 1, 1)),
                dict(size=(5, 1, 1), num_filters=512, pooling_size=(2, 1, 1), pooling_step=(2, 1, 1)),
            ],
            input_shape=patch_shape,
            n_channels=512,
            batch_normalize=hyperparameters["batch_normalize_patch"])