Git Product home page Git Product logo

posec3d-with-attention's Introduction

PoseC3D-with-Attention

Introduced 3D CBAM Attention to the PoseC3D in MMAction2

本Git只提供了code,实际要到https://github.com/open-mmlab/mmaction2 这个git页去clone跑,新py文件替换原本文件

配置教程:https://blog.csdn.net/WhiffeYF/article/details/120556253

针对于一个数据集

先在data下 wget pkl file (annotation file)

再train: 在mmaction2下跑./tools/dist_train.sh ./configs/skeleton/posec3d/slowonly_kinetics400_pretrained_r50_u48_120e_ucf101_split1_keypoint.py 2 --validate

再test :在mmaction2下跑 ./tools/dist_test.sh ./configs/skeleton/posec3d/slowonly_kinetics400_pretrained_r50_u48_120e_ucf101_split1_keypoint.py ./work_dirs/posec3d_iclr/slowonly_kinetics400_pretrained_r50_u48_120e_ucf101_split1_keypoint/best_top1_acc_epoch_11.pth 2 --eval top_k_accuracy

路径都是一样的

ntu60的keypoint和limb各自分开都可以用

topkaccuracy全部公用

benchmark是榜单的意思 (paper with code上找)

heatmap visualization直接在jupyter里跑,下载好pickle就行,放到相应路径

ntu60subtrain是train,val是测试

NTU60: ./tools/dist_train.sh ./configs/skeleton/posec3d/slowonly_r50_u48_240e_ntu60_xsub_keypoint.py 4 --validate

./tools/dist_test.sh ./configs/skeleton/posec3d/slowonly_r50_u48_240e_ntu60_xsub_keypoint.py ./work_dirs/posec3d/slowonly_r50_u48_240e_ntu60_xsub_keypoint/best_top1_acc_epoch_11.pth 2 --eval top_k_accuracy

HMDB: ./tools/dist_train.sh ./configs/skeleton/posec3d/slowonly_kinetics400_pretrained_r50_u48_120e_hmdb51_split1_keypoint.py 2 --validate

./tools/dist_test.sh ./configs/skeleton/posec3d/slowonly_kinetics400_pretrained_r50_u48_120e_hmdb51_split1_keypoint.py ./work_dirs/posec3d_iclr/slowonly_kinetics400_pretrained_r50_u48_120e_hmdb51_split1_keypoint/best_top1_acc_epoch_12.pth 2 --eval top_k_accuracy

为了保持结构的一致性,我把configs里的ucf和hmdb里面stage_blocks改成了(4,6,3). 总共只有三个stage

posec3d-with-attention's People

Contributors

kaan0821 avatar

Stargazers

 avatar  avatar  avatar  avatar qinjiajun avatar 竹摇新桑 avatar  avatar Songzexu avatar  avatar  avatar  avatar

Watchers

 avatar

posec3d-with-attention's Issues

注意力添加位置

您好,我想在第一个stage的第一层添加CBAM,那应该如何进行修改呢
真心希望得到您的回复,谢谢。

注意力位置

你好,大佬。
我看到代码里在第一个stage中不适用注意力,但我在项目里打印了一下模型结构,看到每个stage中都带有注意力,这是怎么回事。
image

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.