RoiAlign aligned=True

Author: jiafatom

torchvision/ops/_register_onnx_ops.py

@@ -25,11 +25,12 @@ def roi_align(g, input, rois, spatial_scale, pooled_height, pooled_width, sampli
         batch_indices = _cast_Long(g, squeeze(g, select(g, rois, 1, g.op('Constant',
                                    value_t=torch.tensor([0], dtype=torch.long))), 1), False)
         rois = select(g, rois, 1, g.op('Constant', value_t=torch.tensor([1, 2, 3, 4], dtype=torch.long)))
+        # TODO: Remove this warning after ONNX opset 16 is supported.
         if aligned:
-            warnings.warn("ONNX export of ROIAlign with aligned=True does not match PyTorch when using malformed boxes,"
-                          " ONNX forces ROIs to be 1x1 or larger.")
-            scale = torch.tensor(0.5 / spatial_scale).to(dtype=torch.float)
-            rois = g.op("Sub", rois, scale)
+            warnings.warn("ROIAlign with aligned=True is not supported in ONNX, but will be supported in opset 16. "
+                          "The workaround is that the user need apply the patch "
+                          "https://github.com/microsoft/onnxruntime/pull/8564 "
+                          "and build ONNXRuntime from source.")
 
         # ONNX doesn't support negative sampling_ratio
         if sampling_ratio < 0: