Skip to content

Latest commit

 

History

History
29 lines (21 loc) · 610 Bytes

README.md

File metadata and controls

29 lines (21 loc) · 610 Bytes

Generator-Classifier Networks

Training Specifications

  1. BATCH_SIZE
  2. NUMBER_OF_EPOCHS
  3. TRAIN_RATIO
  4. RATIO_OF_MASKS
  5. NUMBER_OF_WORKERS (for data loader)

GenClassifier Configuration

  1. MODEL_CODE (genclassifier)
  2. NUMBER_OF_MASK_SAMPLES
  3. PATH_TO_GENERATOR
  4. PATH_TO_CLASSIFIER

Generator Configuration

  1. MODEL_CODE (generator)
  2. IMAGE_CHANNELS IMAGE_HEIGHT IMAGE_WIDTH
  3. ARCHITECTURE (one line for each module)

Classifier Configuration

  1. MODEL_CODE (classifier)
  2. IMAGE_CHANNELS IMAGE_HEIGHT IMAGE_WIDTH
  3. USE_MASKS CROP_SIZE
  4. ARCHITECTURE (one line for each module)