niftynet.network.highres3dnet_small module¶
-
class
niftynet.network.highres3dnet_small.
HighRes3DNetSmall
(num_classes, w_initializer=None, w_regularizer=None, b_initializer=None, b_regularizer=None, acti_func='relu', name='HighRes3DNetSmall')¶ Bases:
niftynet.network.base_net.BaseNet
- implementation of HighRes3DNet:
- Li et al., “On the compactness, efficiency, and representation of 3D convolutional networks: Brain parcellation as a pretext task”, IPMI ‘17
(This is smaller model with an initial stride-2 convolution)
-
layer_op
(images, is_training, layer_id=-1)¶