1 write to bbox_head
Microsoft.ML.TorchSharp (1)
AutoFormerV2\AutoformerV2.cs (1)
44this.bbox_head = new RetinaHead(numClasses);
2 references to bbox_head
Microsoft.ML.TorchSharp (2)
AutoFormerV2\AutoformerV2.cs (2)
82var (classificationInput, regressionInput) = this.bbox_head.forward(fpnOutput); 147bbox_head.Dispose();