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

news/2024/10/30 9:31:11/

测试版本: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))


http://www.ppmy.cn/news/368718.html

相关文章

MMYOLO 自定义数据集从标注到部署保姆级教程

theme: juejin 来自社区 PeterH0323 投稿 AI 已经被应用到各行各业,现如今任何人都可以轻松基于开源框架快速搭建符合自身需求的 AI 应用。本文将基于 MMYOLO 开源框架,基于生活中收集的猫猫数据集,教你如何从零开始训练一个可部署检测模型…

【spring】spring是什么?详解它的特点与模块

作者:Insist-- 个人主页:insist--个人主页 作者会持续更新网络知识和python基础知识,期待你的关注 目录 一、spring介绍 二、spring的特点(七点) 1、简化开发 2、AOP的支持 3、声明式事务的支持 4、方便测试 5、…

LVS负载均衡与DR模式

LVS负载均衡与DR模式 一、DR模式的特点二、LVS-DR中的ARP问题1.VIP地址相同导致响应冲突2.返回报文时源地址使用VIP,导致网关设备的ARP缓存表紊乱 三、DR模式 LVS负载均衡群集部署实验准备实验部署实验步骤1.配置负载调度器(192.168.30.10)2.…

【数据库原理与实践】知识点归纳(上)

第1章 数据库系统概述 (数据处理技术的发展 数据库的基本概念和特点) 一:数据、信息、知识、决策 对应:Data、Information、Knowledge、Decision 二:数据库技术的产生与发展 产生原因:数据管理任务的…

3068说明书

首发本人小站 (https://c.undf.top) AKKO3068说明书 thumbnail: >- date: 2021-06-10 21:20:55 updated: categories: keywords: description: [外链图片转存中…(img-914mL0at-1665060648308)]

有用的

动态导航 CPND车友在线实时路况诞生 实时路况在国内其实并不算新鲜事物。众多国际GPS品牌都试水过实时路况,但在中国大陆都没有取得好的效果。中国实时路况目前最普遍的形式还是基于FM的信息播报,内容无外乎1039路况信息和地图拥堵路段标识。用户对两种方…

StringEscapeUtils.unescapeHtml的使用

富文本提交数据到后台后&#xff0c;保存到数据库的格式可能是这样的&#xff1a; <p>【产品名称】艾酷维多种维生素锌软糖</p> 我们有时候需要的是&#xff1a; <p>【产品名称】艾酷维多种维生素锌软糖</p> 所以就需要用到 StringEscapeUtils 类…

富文本提交数据到后台防止xss攻击

StringEscapeUtils.unescapeHtml的使用 富文本提交数据到后台后&#xff0c;保存到数据库的格式可能是这样的&#xff1a; <p>打发 发顺丰</p> 我们有时候需要的是&#xff1a; <p>打发 发顺丰</p> public static void main(String[] args) { //方式一…