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 GetImageName(self, array, idx):
return b.getImageName(self._handle, array, idx)
def GetImageName(self, array_len):
return b.getImageName(self._handle, array_len)