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: