modify merge

Naoki-Wake · Jun 30, 2022 · d49f901 · d49f901
1 parent 9fee3ec
commit d49f901
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/configs/recognition/arr_tsm2022/tsm_r50_1x1x8_50e_breakfast_rgb.py b/configs/recognition/arr_tsm2022/tsm_r50_1x1x8_50e_breakfast_rgb.py
@@ -6,7 +6,7 @@
     '../../_base_/default_runtime.py'
 ]
 # model settings
-model = dict(cls_head=dict(num_classes=26))#174
+model = dict(cls_head=dict(num_classes=26))#26
 load_from = '/mmaction2/pretrained_models/tsm_r50_256h_1x1x8_50e_sthv2_rgb_20210816-032aa4da.pth'
 # dataset settings
 #dataset_type = 'VideoDataset'
@@ -18,7 +18,7 @@
 dataset_type = 'VideoDataset'
 data_root = 'data/breakfast/videos'
 data_root_val = 'data/breakfast/videos'
-ann_file_train = 'data/breakfast/annotations/wo_pseudo/breakfast_train_list_videos.txt'
+ann_file_train = 'data/breakfast/annotations/wo_pseudo/breakfast_train_list_videos.txt'# with_pseudo_largedatanum
 ann_file_val = 'data/breakfast/annotations/wo_pseudo/breakfast_val_list_videos.txt'
 ann_file_test = 'data/breakfast/annotations/wo_pseudo/breakfast_test_list_videos.txt'
 img_norm_cfg = dict(
@@ -69,7 +69,7 @@
     dict(type='ToTensor', keys=['imgs'])
 ]
 data = dict(
-    videos_per_gpu=4,#6
+    videos_per_gpu=6,#6
     workers_per_gpu=4,#4
     train=dict(
         type=dataset_type,
@@ -95,4 +95,4 @@
     weight_decay=0.0005)
 
 # runtime settings
-work_dir = './work_dirs/experiment20220628_wo_pseudo_onlyheader/'
+work_dir = './work_dirs/experiment20220628_with_pseudo_largedatanum_onlyheader/'
diff --git a/docker/Dockerfile2022 b/docker/Dockerfile2022
@@ -41,4 +41,5 @@ RUN pip install fastapi
 RUN pip install pybsc
 RUN pip install uvicorn
 RUN pip install python-multipart
-RUN pip install seaborn
+RUN pip install seaborn
+RUN pip install scikit-image
diff --git a/mmaction/models/backbones/resnet.py b/mmaction/models/backbones/resnet.py
@@ -499,8 +499,8 @@ def _load_torchvision_checkpoint(self, logger=None):
                                        original_conv_name, loaded_param_names)
                 self._load_bn_params(module.bn, state_dict_torchvision,
                                      original_bn_name, loaded_param_names)
-        import pdb
-        pdb.set_trace()
+        # import pdb
+        # pdb.set_trace()
         # check if any parameters in the 2d checkpoint are not loaded
         remaining_names = set(
             state_dict_torchvision.keys()) - set(loaded_param_names)