Udostępnij za pośrednictwem


AutoFormerV2 Klasa

Definicja

Sieć wykrywania obiektów oparta na sieci szkieletowej AutoFormerV2, która jest wstępnie trenowana tylko w zestawie danych MS COCO. Sieć zawiera 3 skalowanie z różnymi parametrami: mały dla 11 MB, średni dla 21 MB i duży dla 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))
Dziedziczenie
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Konstruktory

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

Inicjuje nowe wystąpienie klasy AutoFormerV2.

Metody

forward(torch+Tensor)

Sieć wykrywania obiektów oparta na sieci szkieletowej AutoFormerV2, która jest wstępnie trenowana tylko w zestawie danych MS COCO. Sieć zawiera 3 skalowanie z różnymi parametrami: mały dla 11 MB, średni dla 21 MB i duży dla 41 MB.

FreezeBN()

Zablokuj wagę usługi BatchNorm2d w sieci.

Dotyczy