Condividi tramite


AutoFormerV2 Classe

Definizione

La rete di rilevamento degli oggetti basata sul backbone AutoFormerV2 pre-sottoposto a training solo nel set di dati MS COCO. La rete contiene 3 scale con parametri diversi: piccole per 11 MB, medie per 21 MB e grandi per 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))
Ereditarietà
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Costruttori

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

Inizializza una nuova istanza della classe AutoFormerV2.

Metodi

forward(torch+Tensor)

La rete di rilevamento degli oggetti basata sul backbone AutoFormerV2 pre-sottoposto a training solo nel set di dati MS COCO. La rete contiene 3 scale con parametri diversi: piccole per 11 MB, medie per 21 MB e grandi per 41 MB.

FreezeBN()

Blocca il peso di BatchNorm2d in rete.

Si applica a