def run(self): print 'BlobFetcher started' while True: db_inds = self._get_next_minibatch_inds() minibatch_db = [self._roidb[i] for i in db_inds] if cfg.TRAIN.USE_OHEM: blobs = get_allrois_minibatch(minibatch_db, self._num_classes) else: blobs = get_minibatch(minibatch_db, self._num_classes) self._queue.put(blobs)
def run(self): print 'BlobFetcher started' while True: db_inds = self._get_next_minibatch_inds() minibatch_db = [self._roidb[i] for i in db_inds] if cfg.TRAIN.USE_OHEM: blobs = get_allrois_minibatch(minibatch_db, self._num_classes) else: blobs = get_minibatch(minibatch_db, self._num_classes) self._queue.put(blobs)
def run(self): print 'BlobFetcher started' while True: if cfg.TRAIN.LABEL_SHUFFLING: cla_inds = self._get_next_classes_inds() db_inds = self._get_next_classes_minibatch_inds(cla_inds) minibatch_db = [self._classes_roidb[cla_inds][i] for i in db_inds] if cfg.TRAIN.USE_OHEM: blobs = get_allrois_minibatch(minibatch_db, self._num_classes) else: blobs = get_minibatch(minibatch_db, self._num_classes) self._queue.put(blobs) # return get_minibatch(minibatch_db, self._num_classes) else: db_inds = self._get_next_minibatch_inds() minibatch_db = [self._roidb[i] for i in db_inds] if cfg.TRAIN.USE_OHEM: blobs = get_allrois_minibatch(minibatch_db, self._num_classes) else: blobs = get_minibatch(minibatch_db, self._num_classes) self._queue.put(blobs)
def _get_next_minibatch(self): """Return the blobs to be used for the next minibatch. If cfg.TRAIN.USE_PREFETCH is True, then blobs will be computed in a separate process and made available through self._blob_queue. """ if cfg.TRAIN.USE_PREFETCH: return self._blob_queue.get() else: db_inds = self._get_next_minibatch_inds() minibatch_db = [self._roidb[i] for i in db_inds] if cfg.TRAIN.USE_OHEM: blobs = get_allrois_minibatch(minibatch_db, self._num_classes) else: blobs = get_minibatch(minibatch_db, self._num_classes) return blobs
def _get_next_minibatch(self): """Return the blobs to be used for the next minibatch. If cfg.TRAIN.USE_PREFETCH is True, then blobs will be computed in a separate process and made available through self._blob_queue. """ if cfg.TRAIN.USE_PREFETCH: return self._blob_queue.get() else: db_inds = self._get_next_minibatch_inds() minibatch_db = [self._roidb[i] for i in db_inds] if cfg.TRAIN.USE_OHEM: blobs = get_allrois_minibatch(minibatch_db, self._num_classes) else: blobs = get_minibatch(minibatch_db, self._num_classes) return blobs