main.py 文件源码

def main():
  args = parse_args()
  config = Config(args)

  # ??????
  os.makedirs(config.output_dir, exist_ok=True)

  # ??????
  model = models.generate_model(config.model)

  # ????????
  img_orig = load_image(config.original_image, [config.width, config.height])
  img_style = load_image(config.style_image, [config.width, config.height] if not config.no_resize_style else None)

  # ?????
  generator = models.Generator(model, img_orig, img_style, config)
  generator.generate(config)