Compartir a través de


AutoFormerV2 Clase

Definición

Red de detección de objetos basada en la red troncal autoformerV2, que solo está entrenada previamente en el conjunto de datos MS COCO. La red contiene 3 escalas con parámetros diferentes: pequeño para 11 MB, medio para 21 MB y grande para 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))
Herencia
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Constructores

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

Inicializa una nueva instancia de la clase AutoFormerV2.

Métodos

Dispose(Boolean)

Red de detección de objetos basada en la red troncal autoformerV2, que solo está entrenada previamente en el conjunto de datos MS COCO. La red contiene 3 escalas con parámetros diferentes: pequeño para 11 MB, medio para 21 MB y grande para 41 MB.

forward(torch+Tensor)

Red de detección de objetos basada en la red troncal autoformerV2, que solo está entrenada previamente en el conjunto de datos MS COCO. La red contiene 3 escalas con parámetros diferentes: pequeño para 11 MB, medio para 21 MB y grande para 41 MB.

FreezeBN()

Inmovilizar el peso de BatchNorm2d en la red.

Se aplica a