From d49f90100c954e340f3dbd93577be0f872b3c0a5 Mon Sep 17 00:00:00 2001 From: Naoki-Wake Date: Thu, 30 Jun 2022 11:18:57 +0900 Subject: [PATCH] modify merge --- .../arr_tsm2022/tsm_r50_1x1x8_50e_breakfast_rgb.py | 8 ++++---- docker/Dockerfile2022 | 3 ++- mmaction/models/backbones/resnet.py | 4 ++-- 3 files changed, 8 insertions(+), 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 index ca488c469a..aff8cff888 100644 --- 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 index ed276de28e..66bbf81988 100644 --- 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 \ No newline at end of file +RUN pip install seaborn +RUN pip install scikit-image \ No newline at end of file diff --git a/mmaction/models/backbones/resnet.py b/mmaction/models/backbones/resnet.py index 9cb24e410b..5ca5b5b365 100644 --- 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)