code: def build_visionmodel(
      name,
      num_classes,
      input_shape=(224, 224, 3),
      freeze_layers=2,
      activation="softmax",
      pt_fc=[],
      pt_ps=[],
  ):
      if name in PREDEFINED_MODELS:
          model = build_predefined(
              name,
              num_classes,
              input_shape=input_shape,
              freeze_layers=freeze_layers,
              activation=activation,
              pt_fc=pt_fc,
              pt_ps=pt_ps,
          )
      elif name == CNN:
          model = build_cnn(
              num_classes,
              input_shape=input_shape,
              activation=activation,
          )
      elif name == WRN22:
          model = create_wide_residual_network(
              input_shape,
              nb_classes=num_classes,
              N=3,
              k=6,
              dropout=0.00,
              activation=activation,
              verbose=0,
          )
      else:
          raise ValueError("Unknown model: %s" % (name))
      U.vprint("%s model created." % (name))
  
      return model
  
file_path: ktrain/ktrain/vision/models.py
imports: from ..None import utils
  from ..imports import *
  from .wrn import create_wide_residual_network
label_desc: Use image node_id 7 for calling a global function with example usage: build_visionmodel(name, num_classes, input_shape, freeze_layers, activation, pt_fc, pt_ps) and returns: model