Поделиться через


AutoFormerV2 Класс

Определение

Сеть обнаружения объектов на основе магистрали AutoFormerV2, которая предварительно обучена только в наборе данных MS COCO. Сеть содержит три шкалы с разными параметрами: маленький для 11 МБ, средний для 21 МБ и большой для 41 МБ.

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))
Наследование
torch.nn.Module<TorchSharp.torch.Tensor,ValueTuple<TorchSharp.torch.Tensor,TorchSharp.torch.Tensor,TorchSharp.torch.Tensor>>
AutoFormerV2

Конструкторы

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

Инициализирует новый экземпляр класса AutoFormerV2.

Методы

forward(torch+Tensor)

Сеть обнаружения объектов на основе магистрали AutoFormerV2, которая предварительно обучена только в наборе данных MS COCO. Сеть содержит три шкалы с разными параметрами: маленький для 11 МБ, средний для 21 МБ и большой для 41 МБ.

FreezeBN()

Заморозите вес BatchNorm2d в сети.

Применяется к