anijain2305 commented on a change in pull request #4417: [TOPI][RELAY][OP] add
op crop_and_resize
URL: https://github.com/apache/incubator-tvm/pull/4417#discussion_r350373294
##########
File path: topi/python/topi/image/resize.py
##########
@@ -210,3 +210,185 @@ def _bicubic(*indices):
raise ValueError('%s method is not supported.' % method)
return tvm.compute(output_shape, compute_func, name='resize',
tag=tag.INJECTIVE)
+
+
+def crop_and_resize(data, boxes, box_indices, crop_size, layout="NCHW",
+ method="bilinear", extrapolation_value=0, out_dtype=None):
+ """Perform crop and resize operation on the data.
+
+ Parameters
+ ----------
+ data : tvm.Tensor
+ inputs is a 4-D tensor with shape
+ [batch, channel, in_height, in_width]
+ or [batch, in_height, in_width, channel]
+
+ boxes : tvm.Tensor
+ A 2-D tensor of shape [num_boxes, 4]. Each row of the tensor specifies
+ the coordinates of a box.
+
+ box_indices : tvm.Tensor
+ A 1-D tensor of shape [num_boxes], box_indices[i] specifies the data
that
+ the i-th box refers to.
+
+ crop_size : Tuple
+ The target size of each box.
+
+ layout : string, optional
+ "NCHW", "NHWC"
+
+ method : {"bilinear", "nearest_neighbor"}
+ Method to be used for resizing.
+
+ extrapolation_value: float, optional
+ Value used for extrapolation, when applicable.
+
+ out_dtype : string, optional
+ Type to return. If left None will be same as input type.
+
+ Returns
+ -------
+ output : tvm.Tensor
+ 4-D with shape [num_boxes, channel, crop_height, crop_width]
+ or [num_boxes, crop_height, crop_width, channel]
+ """
+ method = method.lower()
+ target_h = crop_size[0]
+ target_w = crop_size[1]
+
+ if layout == 'NHWC':
+ output_shape = [box_indices.shape[0], crop_size[0], crop_size[1],
data.shape[3]]
+ image_height = data.shape[1]
+ image_width = data.shape[2]
+ elif layout == 'NCHW':
+ output_shape = [box_indices.shape[0], data.shape[1], crop_size[0],
crop_size[1]]
+ image_height = data.shape[2]
+ image_width = data.shape[3]
+ # Otherwise layout must be NCHWxc
+ else:
+ output_shape = [box_indices.shape[0], data.shape[1],
+ crop_size[0], crop_size[1], data.shape[4]]
+ image_height = data.shape[2]
+ image_width = data.shape[3]
+
+ def _get_pixel(n, c, y, x, cc):
+ if layout.lower() == 'nhwc':
+ return data(n, y.astype("int32"), x.astype("int32"),
c).astype('float')
+ if layout.lower() == 'nchw':
Review comment:
Why not elif? Pylint?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
