AutoFormerV2 Clase

Definición

La red de detección de objetos basada en la red troncal de AutoFormerV2, que solo está entrenada previamente en el conjunto de datos ms COCO. La red contiene 3 escalas con diferentes parámetros: 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

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

Inicializa una nueva instancia de la clase AutoFormerV2.

Métodos

Nombre Description
Dispose(Boolean)

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

forward(torch+Tensor)

La red de detección de objetos basada en la red troncal de AutoFormerV2, que solo está entrenada previamente en el conjunto de datos ms COCO. La red contiene 3 escalas con diferentes parámetros: 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