Method get_result

tensorpack/predict/dataset.py:67–77

(self)

    @HIDE_DOC
    def get_result(self):
        self.dataset.reset_state()
        try:
            sz = len(self.dataset)
        except NotImplementedError:
            sz = 0
        with get_tqdm(total=sz, disable=(sz == 0)) as pbar:
            for dp in self.dataset:
                res = self.predictor(*dp)
                yield res
                pbar.update()


class MultiProcessDatasetPredictor(DatasetPredictorBase):