-
Notifications
You must be signed in to change notification settings - Fork 161
/
Copy pathucf24.yaml
53 lines (53 loc) · 1.6 KB
/
ucf24.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
TRAIN:
RESUME_PATH: "/usr/home/sut/YOWO/backup/ucf24/yowo_ucf24_16f_best_08181.pth"
DATASET: ucf24 # `ava`, `ucf24` or `jhmdb21`
BATCH_SIZE: 10
TOTAL_BATCH_SIZE: 128
LEARNING_RATE: 1e-4
EVALUATE: False
FINE_TUNE: False
BEGIN_EPOCH: 1
END_EPOCH: 10
SOLVER:
MOMENTUM: 0.9
WEIGHT_DECAY: 5e-4
STEPS: [2, 3, 4, 5]
LR_DECAY_RATE: 0.5
ANCHORS: [0.70458, 1.18803, 1.26654, 2.55121, 1.59382, 4.08321, 2.30548, 4.94180, 3.52332, 5.91979]
NUM_ANCHORS: 5
OBJECT_SCALE: 5
NOOBJECT_SCALE: 1
CLASS_SCALE: 1
COORD_SCALE: 1
DATA:
NUM_FRAMES: 16
SAMPLING_RATE: 1
TRAIN_JITTER_SCALES: [256, 320]
TRAIN_CROP_SIZE: 224
TEST_CROP_SIZE: 224
MEAN: [0.4345, 0.4051, 0.3775]
STD: [0.2768, 0.2713, 0.2737]
MODEL:
NUM_CLASSES: 24
BACKBONE_3D: resnext101
BACKBONE_2D: darknet
WEIGHTS:
BACKBONE_3D: "weights/resnext-101-kinetics.pth"
BACKBONE_2D: "weights/yolo.weights"
FREEZE_BACKBONE_3D: False
FREEZE_BACKBONE_2D: False
LISTDATA:
BASE_PTH: "/usr/home/sut/datasets/ucf24"
TRAIN_FILE: "/usr/home/sut/datasets/ucf24/trainlist.txt"
TEST_FILE: "/usr/home/sut/datasets/ucf24/testlist.txt"
TEST_VIDEO_FILE: "/usr/home/sut/datasets/ucf24/testlist_video.txt"
MAX_OBJS: 6
CLASS_NAMES: [
"Basketball", "BasketballDunk", "Biking", "CliffDiving", "CricketBowling",
"Diving", "Fencing", "FloorGymnastics", "GolfSwing", "HorseRiding",
"IceDancing", "LongJump", "PoleVault", "RopeClimbing", "SalsaSpin",
"SkateBoarding", "Skiing", "Skijet", "SoccerJuggling", "Surfing",
"TennisSwing", "TrampolineJumping", "VolleyballSpiking", "WalkingWithDog"
]
BACKUP_DIR: "backup/ucf24"
RNG_SEED: 1