def defineNetwork( self ):
print( 'define network' )
input_tensor = Input( shape = ( self.dim, self.dim,
self.nFlowMaps * self.flowSteps ) )
model = BaseModel( input_tensor = input_tensor,
weights = None,
classes = self.classes )
optimizer = SGD( lr=1e-2, momentum = 0.9, decay=1e-4 )
model.compile( loss = 'categorical_crossentropy',
optimizer = optimizer,
metrics = [ 'acc' ] )
return model
