code: def __init__(
      self,
      output_size,
      data_dir,
      saved_gt_dir,
      mean,
      variance,
      gauss_init_size,
      gauss_sigma,
      enlarge_region,
      enlarge_affinity,
      aug,
      vis_test_dir,
      vis_opt,
      sample,
  ):
      super().__init__(
          output_size,
          data_dir,
          saved_gt_dir,
          mean,
          variance,
          gauss_init_size,
          gauss_sigma,
          enlarge_region,
          enlarge_affinity,
          aug,
          vis_test_dir,
          vis_opt,
          sample,
      )
      self.img_names, self.char_bbox, self.img_words = self.load_data()
      self.vis_index = list(range(1000))
  
imports: import os
  import re
  import itertools
  import random
  import numpy
  import scipy.io
  from PIL import Image
  import cv2
  from torch.utils.data import Dataset
  import torchvision.transforms
  from data import imgproc
  from data.gaussian import GaussianBuilder
  from data.imgaug import rescale, random_resize_crop_synth, random_resize_crop, random_horizontal_flip, random_rotate, random_scale, random_crop
  from data.pseudo_label.make_charbox import PseudoCharBoxBuilder
  from utils.util import saveInput, saveImage
label_desc: Use image node_id 1 to create a new SynthTextDataSet object from inherited base classes: CraftBaseDataset with example: obj = SynthTextDataSet(output_size, data_dir, saved_gt_dir, mean, variance, gauss_init_size, gauss_sigma, enlarge_region, enlarge_affinity, aug, vis_test_dir, vis_opt, sample)