From 3ca196d90542caa9087bbc97518c5b03217abc70 Mon Sep 17 00:00:00 2001 From: nijkah Date: Mon, 23 Sep 2024 15:53:40 +0900 Subject: [PATCH] [Fix] Fix typo in prepare_data --- mmseg/apis/remote_sense_inferencer.py | 4 ++-- mmseg/apis/utils.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mmseg/apis/remote_sense_inferencer.py b/mmseg/apis/remote_sense_inferencer.py index 6726c6ae34..a280060421 100644 --- a/mmseg/apis/remote_sense_inferencer.py +++ b/mmseg/apis/remote_sense_inferencer.py @@ -16,7 +16,7 @@ gdal = None from mmseg.registry import MODELS -from .utils import _preprare_data +from .utils import _prepare_data class RSImage: @@ -224,7 +224,7 @@ def inference(self): self.read_buffer.put(self.END_FLAG) self.write_buffer.put(item) break - data, _ = _preprare_data(item[1], self.model) + data, _ = _prepare_data(item[1], self.model) with torch.no_grad(): result = self.model.test_step(data) item[1] = result[0].pred_sem_seg.cpu().data.numpy()[0] diff --git a/mmseg/apis/utils.py b/mmseg/apis/utils.py index 4cf8775660..7cd8cef6b6 100644 --- a/mmseg/apis/utils.py +++ b/mmseg/apis/utils.py @@ -9,7 +9,7 @@ ImageType = Union[str, np.ndarray, Sequence[str], Sequence[np.ndarray]] -def _preprare_data(imgs: ImageType, model: BaseModel): +def _prepare_data(imgs: ImageType, model: BaseModel): cfg = model.cfg for t in cfg.test_pipeline: