config.yml

DATA:
    DIR: "/content/nfl_resize/"
    VAL_DIR: "/content/nfl_resize_test/"
    TEST_DIR: "/content/image_nfl_s12/image_nfl_s12/"  # /kaggle/working/train_images
    LABEL_FILE: "/content/train_idx_Sideline.txt"
    VAL_LABEL_FILE: "/content/test_idx_Sideline.txt"
    TEST_LABEL_FILE: "/content/test_idx_Sideline.txt"
    # DIR: "data/training/"
    # VAL_DIR: "data/training/"
    # TEST_DIR: "data/training"
    # LABEL_FILE: "data/training/train_1716_100_100.txt"
    # VAL_LABEL_FILE: "data/training/test_1716_100_100.txt"
    # TEST_LABEL_FILE: "data/training/test_1716_100_100.txt"
    BATCH_SIZE: 8
    WORKERS: 4

OPTIMIZER:
    LR: 2e-4
    MOMENTUM: 0.9
    GAMMA: 0.1
    WEIGHT_DECAY: 0
    STEPS: []

MODEL:
    NUMANGLE: 100
    NUMRHO: 100
    FIX: True
    THRESHOLD: 0.01
    EDGE_ALIGN: False
    BACKBONE: resnet50 
    # only support : resnet18, resnet50, resnet101, mobilenetv2, resnext50, res2net50, vgg16

TRAIN:
    EPOCHS: 40
    PRINT_FREQ: 100
    TEST: False
    SEED: 1997
    GPU_ID: 0
    DATA_PARALLEL: False
    RESUME: 

MISC:
    TMP: "/content/result/reproduce"