default_values.yaml

# ###### #
# TRACKS #
# ###### #

track:
    N: 5

tracks:
    inactivity_threshold: 30

# ###################### #
# Dataset and DataLoader #
# ###################### #

dataset:
    crops_folder: ""
    force_img_reshape: false

    pt21:
        check_img_sizes: false
        load_img_crops: true

    kprcnn:
        score_threshold: 0.3
        iou_threshold: 0.6

    MOT:
        space_around_delimiters: False
        file_separator: ",\\s?"
        seqinfo_path: null
        crop_key: "Crops"


dataloader:
    batch_size: 16
    drop_last: false
    shuffle: false
    workers: 0  # right now there is no full multi device support
    collate_fn: null

video_dataset:
    stream: "video"
    num_threads: 0
    video_backend: "pyav"

images:
    force_reshape: false
    image_size: !!python/tuple [1024, 1024]
    image_mode: "zero-pad"
    crop_size: !!python/tuple [256, 256]
    crop_mode: "zero-pad"
    jpeg_quality: 90
    aspect_mode: "zero-pad"
    aspect_round_decimals: 2
    bbox_min_size: 50.0

submission:
    file: "./submission.json"

    MOT:
        bbox_decimals: 3
        score_decimals: 3
        seqinfo_key: "Sequence"

# ############## #
# LOSS FUNCTIONS #
# ############## #

cross_entropy_loss:
    label_smoothing: 0.1

# ####### #
# MODULES #
# ####### #

base:
    print_prio: "INFO"
    description: ""
    log_dir: "./results/"
    log_dir_add_date: true
    log_dir_suffix: ""
    precision: "float32"

alpha:
    act_func: null

engine:
    writer_log_dir_suffix: ""
    writer_kwargs: { }
    
    test:
        normalize: false

    train:
        epochs: 1
        start_epoch: 1
        save_interval: 5
        loss_kwargs: {}
        optim_kwargs:
            lr: 0.0001  # 1e-4
        scheduler: "StepLR"
        scheduler_kwargs:
            step_size: 1
            gamma": 0.1
        load_image_crops: true

    visual:
        topk_acc: [1]
        topk_cmc: [ 1, 5, 10, 20 ]
        metric_kwargs: {}
        write_embeds: [False, False]
        image_key: "image_crop"

    dgs:
        save_images: true
        show_keypoints: true
        show_skeleton: true
        draw_kwargs: {}
        acc_k_train: [ 1, 5, 10, 20, 50 ]
        acc_k_eval: [ 1, 5, 10, 20, 50 ]
        eval_accuracy: true

similarity:
    softmax: true
    
    torchreid:
        sim_kwargs: {}

    oks:
        kp_dim: 2

dgs:
    new_track_weight: 0.0

embed_gen:
    embedding_key: "embedding"
    save_embeddings: true
    
    torchreid:
        weights: "pretrained"
        image_key: "image_crop"

    pose:
        LPBEG:
            bbox_format: "XYWH"
            bias: true
            hidden_layers: []

        KPCPBEG:
            bbox_format: "XYWH"
            bias: true
            nof_kernels: 5
            hidden_layers: []
            hidden_layers_kp: []
            
combine:
    softmax: true