train.yaml

TRAINING:
  # One of "fasterrcnn_resnet50_fpn", "fasterrcnn_mobilenet_v3_large_fpn"
  META_ARCHITECTURE: "fasterrcnn_mobilenet_v3_large_fpn"
  # One of "FasterRCNN_ResNet50_FPN_Weights", "FasterRCNN_MobileNet_V3_Large_FPN_Weights"
  WEIGHTS: "FasterRCNN_MobileNet_V3_Large_FPN_Weights"
  MODEL_FILE: "mobile-netv2.pt"
  NUM_CLASSES: 17
  TEST_PROPORTION: 0.2
  PARAMETERS:
    BASE_LR: 0.001
    MOMENTUM: 0.9
    WEIGHT_DECAY: 0.0001
    BATCH_SIZE: 4
    EPOCHS: 40
    STEPS: 20 #  StepLR only accepts a single number
    GAMMA: 0.1
    NUM_WORKERS: 4
  DEVICE: "cuda"
DATA:
  PATH: "data"
  ANNOTATIONS_PATH: "annotations_all.json"
  SUPERCATEGORIES: ["hauptgebaeude", "karo5", "piloty", "ULB"]
  SUBCATEGORIES: ["right", "back", "left", "front"]
METHOD:
  NAME: "train" # one of execute, train, view, label
WANDB:
  ENTITY: "macaw"
  PROJECT: "augmented-vision"
  IGNORE: True