Share via


AutoFormerV2 Class

Definition

The object detection network based on AutoFormerV2 backbone which is pretrained only in MS COCO dataset. The network contains 3 scales with different parameters: small for 11MB, medium for 21MB and large for 41MB.

public class AutoFormerV2 : TorchSharp.torch.nn.Module<TorchSharp.torch.Tensor,(TorchSharp.torch.Tensor, TorchSharp.torch.Tensor, TorchSharp.torch.Tensor)>
type AutoFormerV2 = class
    inherit torch.nn.Module<torch.Tensor, ValueTuple<torch.Tensor, torch.Tensor, torch.Tensor>>
Public Class AutoFormerV2
Inherits torch.nn.Module(Of torch.Tensor, ValueTuple(Of torch.Tensor, torch.Tensor, torch.Tensor))
Inheritance
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Constructors

AutoFormerV2(Int32, List<Int32>, List<Int32>, List<Int32>, torch+Device)

Initializes a new instance of the AutoFormerV2 class.

Methods

forward(torch+Tensor) TorchSharp.torch.nn.Module`2.forward(TorchSharp.torch.Tensor)
FreezeBN()

Freeze the weight of BatchNorm2d in network.

Applies to