code: def custom_ssd(
      base_network_name,
      base_size,
      filters,
      sizes,
      ratios,
      steps,
      classes,
      dataset,
      pretrained_base,
      **kwargs,
  ):
      """Custom SSD models."""
      if base_network_name == "vgg16_atrous" and base_size == 300:
          features = vgg16_atrous_300
      elif base_network_name == "vgg16_atrous" and base_size == 512:
          features = vgg16_atrous_512
      elif base_network_name == "resnet18_v1" and base_size == 512:
          features = ["stage3_activation1", "stage4_activation1"]
      elif base_network_name == "resnet50_v1" and base_size == 512:
          features = ["stage3_activation5", "stage4_activation2"]
      elif base_network_name == "resnet101_v2" and base_size == 512:
          features = ["stage3_activation22", "stage4_activation2"]
      elif base_network_name == "resnet152_v2" and base_size == 512:
          features = [
              "stage2_activation7",
              "stage3_activation35",
              "stage4_activation2",
          ]
      elif base_network_name == "mobilenet1.0" and base_size == 512:
          features = ["relu22_fwd", "relu26_fwd"]
      elif base_network_name == "mobilenet1.0" and base_size == 300:
          features = ["relu22_fwd", "relu26_fwd"]
      elif base_network_name == "mobilenet0.25" and base_size == 300:
          features = ["relu22_fwd", "relu26_fwd"]
      else:
          raise NotImplementedError(
              "Unsupported network", base_network_name
          )
  
      net = get_ssd(
          name=base_network_name,
          base_size=base_size,
          features=features,
          filters=filters,
          sizes=sizes,
          ratios=ratios,
          steps=steps,
          classes=classes,
          dataset=dataset,
          pretrained_base=pretrained_base,
          **kwargs,
      )
  
      return net
  
imports: from __future__ import absolute_import
  import os
  import warnings
  import mxnet
  from mxnet import autograd
  from mxnet.gluon import nn
  from mxnet.gluon import HybridBlock
  from ...nn.feature import FeatureExpander
  from .anchor import SSDAnchorGenerator
  from ...nn.predictor import ConvPredictor
  from ...nn.coder import MultiPerClassDecoder, NormalizedBoxCenterDecoder
  from .vgg_atrous import vgg16_atrous_300, vgg16_atrous_512
label_desc: Use image node_id 2 for calling a global function with example usage: custom_ssd(base_network_name, base_size, filters, sizes, ratios, steps, classes, dataset, pretrained_base) and returns: net