Freigeben über


AutoFormerV2 Klasse

Definition

Das Objekterkennungsnetzwerk, das auf dem AutoFormerV2-Backbone basiert, der nur im MS COCO-Dataset vortrainiert wird. Das Netzwerk enthält drei Skalierungen mit unterschiedlichen Parametern: klein für 11 MB, mittel für 21 MB und groß für 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))
Vererbung
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Konstruktoren

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

Initialisiert eine neue Instanz der AutoFormerV2-Klasse.

Methoden

forward(torch+Tensor)

Das Objekterkennungsnetzwerk, das auf dem AutoFormerV2-Backbone basiert, der nur im MS COCO-Dataset vortrainiert wird. Das Netzwerk enthält drei Skalierungen mit unterschiedlichen Parametern: klein für 11 MB, mittel für 21 MB und groß für 41 MB.

FreezeBN()

Fixieren sie die Gewichtung von BatchNorm2d im Netzwerk.

Gilt für: