IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

eliminate odd memory copy (#8093)

Browse Source
This commit is contained in:
Vladimir Dudnik 2021-10-28 12:58:08 +03:00 committed by GitHub
parent ab6f9c670e
commit 6ac571e12a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
inference-engine/samples/common/format_reader/opencv_wrapper.cpp
View File

@ -27,22 +27,24 @@ OCVReader::OCVReader(const string& filename) {
}
std::shared_ptr<unsigned char> OCVReader::getData(size_t width = 0, size_t height = 0) {
cv::Mat resized(img);
if (width != 0 && height != 0) {
size_t iw = img.size().width;
size_t ih = img.size().height;
if (width != iw || height != ih) {
slog::warn << "Image is resized from (" << iw << ", " << ih << ") to (" << width << ", " << height << ")"
<< slog::endl;
}
cv::resize(img, resized, cv::Size(width, height));
}
if (width == 0)
width = img.cols;
size_t size = resized.size().width * resized.size().height * resized.channels();
if (height == 0)
height = img.rows;
size_t size = width * height * img.channels();
_data.reset(new unsigned char[size], std::default_delete<unsigned char[]>());
for (size_t id = 0; id < size; ++id) {
_data.get()[id] = resized.data[id];
cv::Mat resized(width, height, img.type(), _data.get());
if (width != img.cols || height != img.rows) {
slog::warn << "Image is resized from (" << img.cols << ", " << img.rows << ") to (" << width << ", " << height
<< ")" << slog::endl;
}
// cv::resize() just copy data to output image if sizes are the same
cv::resize(img, resized, cv::Size(width, height));
return _data;
}
#endif