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

Dispose(Boolean)

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.

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 à