Python ConvNet.add_pooling Exemples

Exemple #1

Fichier : network_train.py Projet : web-engineering-team6/team_6th_application

 def __init__(self):
     super(SimpleConv, self).__init__()
     ConvNet.add_conv(self, 1, 30, 5, 5)
     ConvNet.add_batch_normalization(self, 30 * 24 * 24, "Relu")
     ConvNet.add_pooling(self, 2, 2, stride=2)
     ConvNet.add_affine(self, 30 * 12 * 12, 200)
     ConvNet.add_batch_normalization(self, 200, "Relu")
     ConvNet.add_affine(self, 200, 10)
     ConvNet.add_softmax(self)

Exemple #2

 def __init__(self):
     super(SimpleConv, self).__init__()
     ConvNet.add_conv(self, 3, 64, 3, 3, pad=1)
     ConvNet.add_batch_normalization(self, 64 * 96 * 96, "Relu")
     ConvNet.add_pooling(self, 2, 2, stride=2)
     ConvNet.add_conv(self, 64, 16, 3, 3, pad=1)
     ConvNet.add_batch_normalization(self, 16 * 48 * 48, "Relu")
     ConvNet.add_pooling(self, 2, 2, stride=2)
     ConvNet.add_affine(self, 16 * 24 * 24, 200)
     ConvNet.add_batch_normalization(self, 200, "Relu")
     ConvNet.add_affine(self, 200, 2)
     ConvNet.add_softmax(self)