| 86 | } |
| 87 | |
| 88 | ImageBatchVarShape AverageBlurVarShape(ImageBatchVarShape &input, const std::tuple<int, int> &max_kernel_size, |
| 89 | Tensor &kernel_size, Tensor &kernel_anchor, NVCVBorderType border, |
| 90 | std::optional<Stream> pstream) |
| 91 | { |
| 92 | ImageBatchVarShape output = ImageBatchVarShape::Create(input.capacity()); |
| 93 | |
| 94 | for (int i = 0; i < input.numImages(); ++i) |
| 95 | { |
| 96 | nvcv::ImageFormat format = input[i].format(); |
| 97 | nvcv::Size2D size = input[i].size(); |
| 98 | auto image = Image::Create(size, format); |
| 99 | output.pushBack(image); |
| 100 | } |
| 101 | |
| 102 | return AverageBlurVarShapeInto(output, input, max_kernel_size, kernel_size, kernel_anchor, border, pstream); |
| 103 | } |
| 104 | |
| 105 | } // namespace |
| 106 | |