Partager via


AutoFormerV2 Classe

Définition

Réseau de détection d’objets basé sur la dorsale principale AutoFormerV2, qui est préentraîné uniquement dans le jeu de données MS COCO. Le réseau contient 3 échelles avec différents paramètres : petit pour 11 Mo, moyen pour 21 Mo et grand pour 41 Mo.

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))
Héritage
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Constructeurs

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

Initialise une nouvelle instance de la classe AutoFormerV2.

Méthodes

forward(torch+Tensor)

Réseau de détection d’objets basé sur la dorsale principale AutoFormerV2, qui est préentraîné uniquement dans le jeu de données MS COCO. Le réseau contient 3 échelles avec différents paramètres : petit pour 11 Mo, moyen pour 21 Mo et grand pour 41 Mo.

FreezeBN()

Figer le poids de BatchNorm2d dans le réseau.

S’applique à