search

google-research / deeplab2

DeepLab2 is a TensorFlow library for deep labeling, aiming to provide a unified and state-of-the-art TensorFlow codebase for dense pixel labeling tasks.
Apache License 2.0
998 stars 157 forks source link

Logits and scores of semantic prediction #154

Open sonukiller opened 1 year ago

sonukiller commented 1 year ago

The model output for the semantic predictions is in the format:

semantic logits : (batch, 81, 81, num_classes) (For a crop_size of 321) semantic scores : (batch, img_height, img_width, num_classes).

How to get the semantic logits in the exact shape of semantic scores?