fix a flag

chenxinfeng4 · Aug 4, 2024 · d277e56 · d277e56
1 parent 0ec2701
commit d277e56
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ffmpegcv/ffmpeg_reader_cuda.py b/ffmpegcv/ffmpeg_reader_cuda.py
@@ -205,7 +205,7 @@ def __init__(self, vid:FFmpegReader, gpu=0, tensor_format='hwc'):
             self.converter = load_cuda_module()[(vid.pix_fmt, tensor_format)]
 
     def read(self, out_MAT:gpuarray.GPUArray=None) -> Tuple[bool, gpuarray.GPUArray]:
-        self.waitInit = True
+        self.waitInit = False
         ret, frame_yuv420p = self.vid.read()
         if not ret:
             return False, None
@@ -219,7 +219,7 @@ def read(self, out_MAT:gpuarray.GPUArray=None) -> Tuple[bool, gpuarray.GPUArray]
             return True, out_MAT
 
     def read_cudamem(self, out_MAT:cuda.DeviceAllocation=None) -> Tuple[bool, cuda.DeviceAllocation]:
-        self.waitInit = True
+        self.waitInit = False
         ret, frame_yuv420p = self.vid.read()
         if not ret:
             return False, None
@@ -235,7 +235,7 @@ def read_cudamem(self, out_MAT:cuda.DeviceAllocation=None) -> Tuple[bool, cuda.D
 
     def read_torch(self, out_MAT=None):
         import torch
-        self.waitInit = True
+        self.waitInit = False
         ret, frame_yuv420p = self.vid.read()
         if not ret:
             return False, None