def next(self):
if self._index >= len(self._batches):
raise StopIteration
s = slice(*(self._batches[self._index]))
Xbuf = self._X[s,:]
if Xbuf.dtype == 'uint8':
if self._Xbuf == None:
self._Xbuf = bm.empty((self._batchsize,self._X.shape[1]))
self._Xbuf[:s.stop-s.start,:] = Xbuf # copy
Xbuf = self._Xbuf[:s.stop-s.start,:] # point to copy
bm.imul(Xbuf,1./255)
batch = DataFold(Xbuf,self._Y[s,:],self._S[s,:] if self._S != None else None)
self._index += 1
return batch
def next(self):
if self._index >= len(self._batches):
raise StopIteration
s = slice(*(self._batches[self._index]))
Xbuf = self._X[s, :]
if Xbuf.dtype == 'uint8':
if self._Xbuf == None:
self._Xbuf = bm.empty((self._batchsize, self._X.shape[1]))
self._Xbuf[:s.stop - s.start, :] = Xbuf # copy
Xbuf = self._Xbuf[:s.stop - s.start, :] # point to copy
bm.imul(Xbuf, 1. / 255)
batch = DataFold(Xbuf, self._Y[s, :],
self._S[s, :] if self._S != None else None)
self._index += 1
return batch
