RoIAlign

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

見 roi_align()。

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

定義每次呼叫時執行的計算。

應被所有子類重寫。

注意

儘管 forward pass 的實現需要在本函式中定義，但後續應呼叫 Module 例項，而不是直接呼叫此函式，因為前者會處理註冊的鉤子（hooks），而後者會靜默忽略它們。