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)