快捷方式

RoIAlign

class torchvision.ops.RoIAlign(output_size: None, spatial_scale: float, sampling_ratio: int, aligned: bool = False)[source]

参见 roi_align().

forward(input: Tensor, rois: Union[Tensor, List[Tensor]]) Tensor[source]

定义每次调用时执行的计算。

应由所有子类覆盖。

注意

尽管正向传递的配方需要在此函数内定义,但应随后调用Module 实例,而不是此函数,因为前者负责运行注册的钩子,而后者会静默地忽略它们。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源