dltk.models.classification package

dltk.models.classification.lenet module

class dltk.models.classification.lenet.LeNet5(num_classes=10, name='lenet5')

Bases: dltk.core.modules.base.AbstractModule

LeNet5 classification network according to

dltk.models.classification.resnet module

class dltk.models.classification.resnet.ResNet(num_classes, num_residual_units=5, filters=[16, 16, 32, 64], strides=[[1, 1, 1], [1, 1, 1], [2, 2, 2], [2, 2, 2]], relu_leakiness=0.01, name='resnet32')

Bases: dltk.core.modules.base.AbstractModule

ResNet module

This module builds a ResNet for classification according to He et al. 2015

Module contents