def __next__(self):
if b.getRemainingImages(self.loader._handle) < self.bs:
raise StopIteration
if self.loader.run() != 0:
raise StopIteration
self.loader.copyImage(self.out_image)
if ((self.loader._name == "Caffe2ReaderDetection")
or (self.loader._name == "CaffeReaderDetection")):
for i in range(self.bs):
size = b.getImageNameLen(self.loader._handle, i)
print(size)
self.array = np.array([" "])
self.out = np.frombuffer(self.array, dtype=(self.array).dtype)
b.getImageName(self.loader._handle, self.out, i)
return self.out_image, self.out_bbox, self.out_tensor
else:
return self.out_image, self.out_tensor
def GetImageNameLength(self, idx):
return b.getImageNameLen(self._handle, idx)
def GetImageNameLen(self, array):
return b.getImageNameLen(self._handle, array)
