bitorch.models.resnet

Classes

BasicBlockV1(in_channels, out_channels, stride)

BasicBlock V1 from "Deep Residual Learning for Image Recognition" paper.

BasicBlockV2(in_channels, out_channels, stride)

BasicBlock V2 from "Identity Mappings in Deep Residual Networks" paper.

BottleneckV1(in_channels, out_channels, stride)

Bottleneck V1 from "Deep Residual Learning for Image Recognition" paper.

BottleneckV2(in_channels, out_channels, stride)

ResNetV1(block, layers, channels, classes[, ...])

ResNet V1 model from "Deep Residual Learning for Image Recognition" paper.

ResNetV2(block, layers, channels[, classes, ...])

ResNet V2 model from "Identity Mappings in Deep Residual Networks" paper.

Resnet(resnet_version, resnet_num_layers, ...)

Resnet152V1(*args, **kwargs)

ResNet-152 V1 model from "Deep Residual Learning for Image Recognition" paper.

Resnet152V2(*args, **kwargs)

ResNet-152 V2 model from "Deep Residual Learning for Image Recognition" paper.

Resnet18V1(*args, **kwargs)

ResNet-18 V1 model from "Deep Residual Learning for Image Recognition" paper.

Resnet18V2(*args, **kwargs)

ResNet-18 V2 model from "Deep Residual Learning for Image Recognition" paper.

Resnet34V1(*args, **kwargs)

ResNet-34 V1 model from "Deep Residual Learning for Image Recognition" paper.

Resnet34V2(*args, **kwargs)

ResNet-34 V2 model from "Deep Residual Learning for Image Recognition" paper.

Resnet50V1(*args, **kwargs)

ResNet-50 V1 model from "Deep Residual Learning for Image Recognition" paper.

Resnet50V2(*args, **kwargs)

ResNet-50 V2 model from "Deep Residual Learning for Image Recognition" paper.

SpecificResnet(classes, channels)

Superclass for ResNet models