mirror of
https://github.com/zebrajr/opencv.git
synced 2025-12-06 12:19:50 +01:00
Warning fixes in DLpack integration on Win32.
This commit is contained in:
parent
e9bded6ff3
commit
80fef356ec
|
|
@ -112,7 +112,7 @@ bool parseDLPackTensor(DLManagedTensor* tensor, cv::cuda::GpuMat& obj, bool copy
|
||||||
static_cast<int>(tensor->dl_tensor.shape[1]),
|
static_cast<int>(tensor->dl_tensor.shape[1]),
|
||||||
type,
|
type,
|
||||||
tensor->dl_tensor.data,
|
tensor->dl_tensor.data,
|
||||||
tensor->dl_tensor.strides[0] * tensor->dl_tensor.dtype.bits / 8
|
static_cast<size_t>(tensor->dl_tensor.strides[0] * tensor->dl_tensor.dtype.bits / 8)
|
||||||
);
|
);
|
||||||
if (copy)
|
if (copy)
|
||||||
obj = obj.clone();
|
obj = obj.clone();
|
||||||
|
|
@ -140,7 +140,7 @@ bool parseDLPackTensor(DLManagedTensor* tensor, cv::cuda::GpuMatND& obj, bool co
|
||||||
std::vector<int> sizes(tensor->dl_tensor.ndim);
|
std::vector<int> sizes(tensor->dl_tensor.ndim);
|
||||||
for (int i = 0; i < tensor->dl_tensor.ndim - 1; ++i)
|
for (int i = 0; i < tensor->dl_tensor.ndim - 1; ++i)
|
||||||
{
|
{
|
||||||
steps[i] = tensor->dl_tensor.strides[i] * tensor->dl_tensor.dtype.bits / 8;
|
steps[i] = static_cast<size_t>(tensor->dl_tensor.strides[i] * tensor->dl_tensor.dtype.bits / 8);
|
||||||
sizes[i] = static_cast<int>(tensor->dl_tensor.shape[i]);
|
sizes[i] = static_cast<int>(tensor->dl_tensor.shape[i]);
|
||||||
}
|
}
|
||||||
sizes.back() = static_cast<int>(tensor->dl_tensor.shape[tensor->dl_tensor.ndim - 1]);
|
sizes.back() = static_cast<int>(tensor->dl_tensor.shape[tensor->dl_tensor.ndim - 1]);
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user