build(config): 更新

ZJCV · Oct 4, 2020 · abdd613 · abdd613
1 parent 1a57a93
commit abdd613
Show file tree

Hide file tree

Showing 6 changed files with 27 additions and 15 deletions.
diff --git a/configs/tsn_r50_hmdb51_rgb_224x224x3.yaml b/configs/tsn_r50_hmdb51_rgb_224x224x3.yaml
@@ -12,8 +12,9 @@ DATASETS:
     ANNOTATION_DIR: 'data/hmdb51'
     SPLIT: 1
 TRANSFORM:
-  INPUT_SIZE: (224, 224, 3)
-  RANDOM_RESIZE_RANGE: (256, 320)
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 256
+  JITTER_SCALES: (256, 320)
   MEAN: (0.485, 0.456, 0.406)
   STD: (0.229, 0.224, 0.225)
 DATALOADER:
@@ -23,7 +24,8 @@ DATALOADER:
 MODEL:
   NAME: 'TSN'
   PRETRAINED: ""
-  SYNC_BN: True
+  SYNC_BN: False
+  INPUT_SIZE: (224, 224, 3)
   BACKBONE:
     NAME: 'resnet50'
     PARTIAL_BN: False

diff --git a/configs/tsn_r50_hmdb51_rgbdiff_224x224x3.yaml b/configs/tsn_r50_hmdb51_rgbdiff_224x224x3.yaml
@@ -12,8 +12,9 @@ DATASETS:
     ANNOTATION_DIR: 'data/hmdb51'
     SPLIT: 1
 TRANSFORM:
-  INPUT_SIZE: (224, 224, 3)
-  RANDOM_RESIZE_RANGE: (256, 320)
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 256
+  JITTER_SCALES: (256, 320)
   MEAN: (0.485, 0.456, 0.406)
   STD: (0.229, 0.224, 0.225)
 DATALOADER:
@@ -24,6 +25,7 @@ MODEL:
   NAME: 'TSN'
   PRETRAINED: ""
   SYNC_BN: True
+  INPUT_SIZE: (224, 224, 3)
   BACKBONE:
     NAME: 'resnet50'
     PARTIAL_BN: False

diff --git a/configs/tsn_r50_jester_rgb_224x224x3.yaml b/configs/tsn_r50_jester_rgb_224x224x3.yaml
@@ -10,8 +10,9 @@ DATASETS:
     DATA_DIR: 'data/jester/20bn-jester-v1'
     ANNOTATION_DIR: 'data/jester'
 TRANSFORM:
-  INPUT_SIZE: (224, 224, 3)
-  RANDOM_RESIZE_RANGE: (256, 320)
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 256
+  JITTER_SCALES: (256, 320)
   MEAN: (0.485, 0.456, 0.406)
   STD: (0.229, 0.224, 0.225)
 DATALOADER:
@@ -21,7 +22,8 @@ DATALOADER:
 MODEL:
   NAME: 'TSN'
   PRETRAINED: ""
-  SYNC_BN: True
+  SYNC_BN: False
+  INPUT_SIZE: (224, 224, 3)
   BACKBONE:
     NAME: 'resnet50'
     PARTIAL_BN: False

diff --git a/configs/tsn_r50_jester_rgbdiff_224x224x3.yaml b/configs/tsn_r50_jester_rgbdiff_224x224x3.yaml
@@ -10,8 +10,9 @@ DATASETS:
     DATA_DIR: 'data/jester/20bn-jester-v1'
     ANNOTATION_DIR: 'data/jester'
 TRANSFORM:
-  INPUT_SIZE: (224, 224, 3)
-  RANDOM_RESIZE_RANGE: (256, 320)
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 256
+  JITTER_SCALES: (256, 320)
   MEAN: (0.485, 0.456, 0.406)
   STD: (0.229, 0.224, 0.225)
 DATALOADER:
@@ -22,6 +23,7 @@ MODEL:
   NAME: 'TSN'
   PRETRAINED: ""
   SYNC_BN: True
+  INPUT_SIZE: (224, 224, 3)
   BACKBONE:
     NAME: 'resnet50'
     PARTIAL_BN: False

diff --git a/configs/tsn_r50_ucf101_rgb_224x224x3.yaml b/configs/tsn_r50_ucf101_rgb_224x224x3.yaml
@@ -12,8 +12,9 @@ DATASETS:
     ANNOTATION_DIR: 'data/ucf101'
     SPLIT: 1
 TRANSFORM:
-  INPUT_SIZE: (224, 224, 3)
-  RANDOM_RESIZE_RANGE: (256, 320)
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 256
+  JITTER_SCALES: (256, 320)
   MEAN: (0.485, 0.456, 0.406)
   STD: (0.229, 0.224, 0.225)
 DATALOADER:
@@ -23,7 +24,8 @@ DATALOADER:
 MODEL:
   NAME: 'TSN'
   PRETRAINED: ""
-  SYNC_BN: True
+  SYNC_BN: False
+  INPUT_SIZE: (224, 224, 3)
   BACKBONE:
     NAME: 'resnet50'
     PARTIAL_BN: False

diff --git a/configs/tsn_r50_ucf101_rgbdiff_224x224x3.yaml b/configs/tsn_r50_ucf101_rgbdiff_224x224x3.yaml
@@ -12,8 +12,9 @@ DATASETS:
     ANNOTATION_DIR: 'data/ucf101'
     SPLIT: 1
 TRANSFORM:
-  INPUT_SIZE: (224, 224, 3)
-  RANDOM_RESIZE_RANGE: (256, 320)
+  TRAIN_CROP_SIZE: 224
+  TEST_CROP_SIZE: 256
+  JITTER_SCALES: (256, 320)
   MEAN: (0.485, 0.456, 0.406)
   STD: (0.229, 0.224, 0.225)
 DATALOADER:
@@ -24,6 +25,7 @@ MODEL:
   NAME: 'TSN'
   PRETRAINED: ""
   SYNC_BN: True
+  INPUT_SIZE: (224, 224, 3)
   BACKBONE:
     NAME: 'resnet50'
     PARTIAL_BN: False