[Fix] Fix inputs dtype convert in data preprocessor (open-mmlab#1975)

imclab · Aug 26, 2022 · ef2152f · ef2152f
1 parent 8de0050
commit ef2152f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mmseg/models/data_preprocessor.py b/mmseg/models/data_preprocessor.py
@@ -104,10 +104,9 @@ def forward(self, data: dict, training: bool = False) -> Dict[str, Any]:
         if self.channel_conversion and inputs[0].size(0) == 3:
             inputs = [_input[[2, 1, 0], ...] for _input in inputs]
 
+        inputs = [_input.float() for _input in inputs]
         if self._enable_normalize:
             inputs = [(_input - self.mean) / self.std for _input in inputs]
-        else:
-            inputs = [_input.float() for _input in inputs]
 
         if training:
             assert data_samples is not None, ('During training, ',