[深度学习][原创]mmaction2预测结果top5得分改为0-1

测试版本：mmaction0.20.0

测试环境:ubuntu18.04

当我们测试tsn模型：

python demo/demo.py configs/recognition/tsn/tsn_r50_video_inference_1x1x3_100e_kinetics400_rgb.py checkpoints/tsn_r50_1x1x3_100e_kinetics400_rgb_20200614-e508be42.pth demo/demo.mp4 tools/data/kinetics/label_map_k400.txt

结果显示：

load checkpoint from local path: checkpoints/tsn_r50_1x1x3_100e_kinetics400_rgb_20200614-e508be42.pth The top-5 labels with corresponding scores are: arm wrestling:  29.61644 rock scissors paper:  10.754842 shaking hands:  9.908401 clapping:  9.189913 massaging feet:  8.305306 显然上面得分>1不符合0-1概率，因此如何改为0-1概率显示呢。

方法如下：

经过测试发现tsn_r50_video_inference_1x1x3_100e_kinetics400_rgb.py加入test_cfg已经没有效果，正确做法就是：

去mmaction2-0.20.0/configs/_base_/models/tsn_r50.py将tesct_cfg改为

test_cfg = dict(average_clips='prob', test_crops=1))

最终文件内容结果：

# model settings model = dict(     type='Recognizer2D',     backbone=dict(         type='ResNet',         pretrained='torchvision://resnet50',         depth=50,         norm_eval=False),     cls_head=dict(         type='TSNHead',         num_classes=400,         in_channels=2048,         spatial_type='avg',         consensus=dict(type='AvgConsensus', dim=1),         dropout_ratio=0.4,         init_std=0.01),     # model training and testing settings     train_cfg=None,     test_cfg = dict(average_clips='prob', test_crops=1))