multi_tsn_frame.yaml 916 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
MODEL:
    name: "TSN"
    format: "frames" # support for "frames" or "videos"
    num_classes: 101
    seg_num: 3
    seglen: 1
    image_mean: [0.485, 0.456, 0.406]
    image_std: [0.229, 0.224, 0.225]
    num_layers: 50
    topk: 5

TRAIN:
    epoch: 80
    short_size: 256
    target_size: 224
    batch_size: 128
    use_gpu: True
    filelist: "./data/dataset/ucf101/ucf101_train_split_1_rawframes.txt"
    learning_rate: 0.001
    learning_rate_decay: 0.1
    decay_epochs: [30, 60]
    l2_weight_decay: 1e-4
    momentum: 0.9
    total_videos: 9738 
25 26
    num_workers: 4
    use_shuffle: True
27 28 29 30 31 32

VALID:
    short_size: 256
    target_size: 224
    batch_size: 128
    filelist: "./data/dataset/ucf101/ucf101_val_split_1_rawframes.txt"
33
    num_workers: 4
34 35 36 37

TEST:
    short_size: 256
    target_size: 224
38 39
    batch_size: 128
    filelist: "./data/dataset/ucf101/ucf101_val_split_1_rawframes.txt"
40
    num_workers: 4