LibMTL.model.resnet_dilated

class ResnetDilated(orig_resnet, dilate_scale=8)[source]

Bases: torch.nn.Module

forward(self, x)[source]
resnet_dilated(basenet, pretrained=True, dilate_scale=8)[source]

Dilated Residual Network models from “Dilated Residual Networks”

Parameters

  • basenet (str) – The type of ResNet.

  • pretrained (bool) – If True, returns a model pre-trained on ImageNet.

  • dilate_scale ({8, 16}, default=8) – The type of dilating process.