AutoFormerV2 Classe

Definição

Namespace:

Microsoft.ML.TorchSharp.AutoFormerV2

Assembly:

Microsoft.ML.TorchSharp.dll

Pacote:

Microsoft.ML.TorchSharp v0.21.1

A rede de detecção de objetos com base no backbone AutoFormerV2, que é pré-treinado somente no conjunto de dados MS COCO. A rede contém três escalas com parâmetros diferentes: pequena para 11 MB, média para 21 MB e grande para 41 MB.

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))

Herança

torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>

AutoFormerV2

Construtores

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

Inicializa uma nova instância da classe AutoFormerV2.

Métodos

forward(torch+Tensor)	A rede de detecção de objetos com base no backbone AutoFormerV2, que é pré-treinado somente no conjunto de dados MS COCO. A rede contém três escalas com parâmetros diferentes: pequena para 11 MB, média para 21 MB e grande para 41 MB.
FreezeBN()	Congele o peso de BatchNorm2d na rede.