From 304102fb1018274a6ff8d7299e1562d3e2cf74ee Mon Sep 17 00:00:00 2001 From: chengjuzhou Date: Mon, 17 May 2021 19:41:44 +0800 Subject: [PATCH 1/2] correct the size of returne tensor --- torchvision/ops/ps_roi_align.py | 2 +- torchvision/ops/ps_roi_pool.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/ops/ps_roi_align.py b/torchvision/ops/ps_roi_align.py index 8097211dfe9..d27f8bbe656 100644 --- a/torchvision/ops/ps_roi_align.py +++ b/torchvision/ops/ps_roi_align.py @@ -39,7 +39,7 @@ def ps_roi_align( ``ceil(roi_width / output_width)``, and likewise for height). Default: -1 Returns: - Tensor[K, C, output_size[0], output_size[1]]: The pooled RoIs + Tensor[K, C/(output_size[0]*output_size[1]), output_size[0], output_size[1]]: The pooled RoIs """ _assert_has_ops() check_roi_boxes_shape(boxes) diff --git a/torchvision/ops/ps_roi_pool.py b/torchvision/ops/ps_roi_pool.py index 01cee261f77..c393a458126 100644 --- a/torchvision/ops/ps_roi_pool.py +++ b/torchvision/ops/ps_roi_pool.py @@ -33,7 +33,7 @@ def ps_roi_pool( the box coordinates. Default: 1.0 Returns: - Tensor[K, C, output_size[0], output_size[1]]: The pooled RoIs. + Tensor[K, C/(output_size[0]*output_size[1]), output_size[0], output_size[1]]: The pooled RoIs. """ _assert_has_ops() check_roi_boxes_shape(boxes) From ad79608bfe43b640211894ca597b8947cf9395f1 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Mon, 17 May 2021 16:12:39 +0000 Subject: [PATCH 2/2] Styling changes --- torchvision/ops/ps_roi_align.py | 2 +- torchvision/ops/ps_roi_pool.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/ops/ps_roi_align.py b/torchvision/ops/ps_roi_align.py index d27f8bbe656..d42353e2b0d 100644 --- a/torchvision/ops/ps_roi_align.py +++ b/torchvision/ops/ps_roi_align.py @@ -39,7 +39,7 @@ def ps_roi_align( ``ceil(roi_width / output_width)``, and likewise for height). Default: -1 Returns: - Tensor[K, C/(output_size[0]*output_size[1]), output_size[0], output_size[1]]: The pooled RoIs + Tensor[K, C / (output_size[0] * output_size[1]), output_size[0], output_size[1]]: The pooled RoIs """ _assert_has_ops() check_roi_boxes_shape(boxes) diff --git a/torchvision/ops/ps_roi_pool.py b/torchvision/ops/ps_roi_pool.py index c393a458126..d0331e557fd 100644 --- a/torchvision/ops/ps_roi_pool.py +++ b/torchvision/ops/ps_roi_pool.py @@ -33,7 +33,7 @@ def ps_roi_pool( the box coordinates. Default: 1.0 Returns: - Tensor[K, C/(output_size[0]*output_size[1]), output_size[0], output_size[1]]: The pooled RoIs. + Tensor[K, C / (output_size[0] * output_size[1]), output_size[0], output_size[1]]: The pooled RoIs. """ _assert_has_ops() check_roi_boxes_shape(boxes)