AutoFormerV2 Class
Definition
Important
Some information relates to prerelease product that may be substantially modified before it’s released. Microsoft makes no warranties, express or implied, with respect to the information provided here.
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. |