MCPcopy Create free account
hub / github.com/NVlabs/SegFormer / whole_inference

Method whole_inference

mmseg/models/segmentors/encoder_decoder.py:214–226  ·  view source on GitHub ↗

Inference with full image.

(self, img, img_meta, rescale)

Source from the content-addressed store, hash-verified

212 return preds
213
214 def whole_inference(self, img, img_meta, rescale):
215 """Inference with full image."""
216
217 seg_logit = self.encode_decode(img, img_meta)
218 if rescale:
219 seg_logit = resize(
220 seg_logit,
221 size=img_meta[0]['ori_shape'][:2],
222 mode='bilinear',
223 align_corners=self.align_corners,
224 warning=False)
225
226 return seg_logit
227
228 def inference(self, img, img_meta, rescale):
229 """Inference with slide/whole style.

Callers 1

inferenceMethod · 0.95

Calls 2

encode_decodeMethod · 0.95
resizeFunction · 0.90

Tested by

no test coverage detected