add init

Author: haoquan

config/X3D_M.yaml

@@ -0,0 +1,60 @@
+TRAIN:
+  # ENABLE: False # default True
+  DATASET: kinetics
+  BATCH_SIZE: 16
+  EVAL_PERIOD: 10
+  CHECKPOINT_PERIOD: 10
+  AUTO_RESUME: True
+X3D:
+  WIDTH_FACTOR: 2.0
+  DEPTH_FACTOR: 2.2
+  BOTTLENECK_FACTOR: 2.25
+  DIM_C5: 2048
+  DIM_C1: 12
+TEST:
+  ENABLE: True
+  DATASET: kinetics
+  BATCH_SIZE: 64
+  # CHECKPOINT_FILE_PATH: 'x3d_m.pyth' # 76.21% top1 30-view accuracy to download from the model zoo (optional).
+  # NUM_SPATIAL_CROPS: 1
+  NUM_SPATIAL_CROPS: 3
+DATA:
+  PATH_TO_DATA_DIR: /home/tione/notebook/video/datafile  ## !!!!
+  NUM_FRAMES: 16
+  SAMPLING_RATE: 50 ## !!!!
+  TARGET_FPS: 25  ## !!!!
+  TRAIN_JITTER_SCALES: [256, 320]
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 224 # use if TEST.NUM_SPATIAL_CROPS: 1
+#   TEST_CROP_SIZE: 256 # use if TEST.NUM_SPATIAL_CROPS: 3
+  INPUT_CHANNEL_NUM: [3]
+  DECODING_BACKEND: pyav    ## !!!!
+RESNET:
+  ZERO_INIT_FINAL_BN: True
+  TRANS_FUNC: x3d_transform
+  STRIDE_1X1: False
+BN:
+  USE_PRECISE_STATS: True
+  NUM_BATCHES_PRECISE: 200
+  WEIGHT_DECAY: 0.0
+SOLVER:
+  BASE_LR: 0.1 # 1 machine
+  BASE_LR_SCALE_NUM_SHARDS: True
+  LR_POLICY: cosine
+  MAX_EPOCH: 10
+  WEIGHT_DECAY: 5e-5
+  WARMUP_EPOCHS: 35.0
+  WARMUP_START_LR: 0.01
+  OPTIMIZING_METHOD: sgd
+MODEL:
+  NUM_CLASSES: 82  ## !!!!
+  ARCH: x3d
+  MODEL_NAME: X3D
+  LOSS_FUNC: cross_entropy
+  DROPOUT_RATE: 0.5
+DATA_LOADER:
+  NUM_WORKERS: 8
+  PIN_MEMORY: True
+NUM_GPUS: 1   ## !!!!
+RNG_SEED: 0
+OUTPUT_DIR: .

config/cfg.py

@@ -0,0 +1,23 @@
+from slowfast.config.defaults import get_cfg
+
+# 可以在defauls中查看所有参数
+def loading_config(cfg_file):
+    # Setup cfg.
+    cfg = get_cfg()
+    # Load config from file.
+    if cfg_file is not None:
+        cfg.merge_from_file(cfg_file)
+
+    cfg.NUM_SHARDS = 1
+    cfg.SHARD_ID = 0
+
+    add_custom_config(cfg)
+    
+    return cfg
+
+
+
+# 可以在这里定义新的参数
+def add_custom_config(cfg):
+    # Add your own customized configs.
+    pass