code: def __init__(self, net, output_op):
      super().__init__()
      self.net = net
      self.output_op = output_op
      assert output_op in [
          "argmax",
          "softmax",
      ], "output_op should in ['argmax', 'softmax']"
  
file_path: PaddleSeg/contrib/CityscapesSOTA/export.py
imports: import argparse
  import os
  import paddle
  import yaml
  from paddleseg.cvlibs import Config, SegBuilder
  from paddleseg.utils import logger
  import datasets, models
label_desc: Use image node_id 1 to create a new SavedSegmentationNet object from inherited base classes: paddle.nn with example: obj = SavedSegmentationNet(net, output_op)