talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make uploading to GPU optional in Image.GetGpuBuffer().

Browse Source 
PiperOrigin-RevId: 529066617
This commit is contained in:
MediaPipe Team
2023-05-03 05:57:58 -07:00
committed by Copybara-Service
parent 3789156a41
commit baa8fc68a1
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/formats/image.h
+5 -5
View File
@@ -113,11 +113,11 @@ class Image {
#endif // MEDIAPIPE_GPU_BUFFER_USE_CV_PIXEL_BUFFER
#endif // !MEDIAPIPE_DISABLE_GPU
// Get a GPU view. Automatically uploads from CPU if needed.
const mediapipe::GpuBuffer GetGpuBuffer() const {
#if !MEDIAPIPE_DISABLE_GPU
if (use_gpu_ == false) ConvertToGpu();
#endif // !MEDIAPIPE_DISABLE_GPU
// Provides access to the underlying GpuBuffer storage.
// Automatically uploads from CPU to GPU if needed and requested through the
// `upload_to_gpu` argument.
const mediapipe::GpuBuffer GetGpuBuffer(bool upload_to_gpu = true) const {
if (!use_gpu_ && upload_to_gpu) ConvertToGpu();
return gpu_buffer_;
}