https://github.com/zjcv/non-local
[CVPR 2018] Non-local Neural Networks
https://github.com/zjcv/non-local
action-recognition c2d i3d non-local pytorch resnet3d video-recognition video-understanding
Last synced: 2 months ago
JSON representation
[CVPR 2018] Non-local Neural Networks
- Host: GitHub
- URL: https://github.com/zjcv/non-local
- Owner: ZJCV
- License: apache-2.0
- Created: 2020-09-26T15:53:05.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2020-12-15T12:16:16.000Z (over 4 years ago)
- Last Synced: 2025-03-04T19:02:13.552Z (3 months ago)
- Topics: action-recognition, c2d, i3d, non-local, pytorch, resnet3d, video-recognition, video-understanding
- Language: Python
- Homepage:
- Size: 6.17 MB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
«Non-local»复现了论文Non-local Neural Networks提出的视频分类模型
* `CodeBase`：[ZJCV/TSN](https://github.com/ZJCV/TSN)和[ZJCV/ZCls](https://github.com/ZJCV/ZCls)
## 内容列表
- [内容列表](#内容列表)
- [背景](#背景)
- [安装](#安装)
- [使用](#使用)
- [主要维护人员](#主要维护人员)
- [致谢](#致谢)
- [参与贡献方式](#参与贡献方式)
- [许可证](#许可证)
## 背景
受到传统图像处理算法`non-local means`影响，[Non-local Neural Networks](https://arxiv.org/abs/1608.00859)提出了注意力模块`Non-local Block`
## 安装
通过`requirements.txt`安装运行所需依赖
```
$ pip install -r requirements.txt
```
处理数据时需要额外安装[denseflow](https://github.com/open-mmlab/denseflow)，可以在[innerlee/setup](https://github.com/innerlee/setup)中找到安装脚本
## 使用
首先设置`GPU`和当前位置
```
$ export CUDA_VISIBLE_DEVICES=1
$ export PYTHONPATH=.
```
* 训练
```
# 训练UCF101
# 单GPU
$ python tools/train.py --config_file=configs/i3d-3x3-nl_r3d50_ucf101_rgb_224x32_dense.yaml
# 多GPU
$ python tools/train.py \
--config_file=configs/i3d-3x3-nl_r3d50_ucf101_rgb_224x32_dense.yaml \
--eval_step=1000 \
--save_step=1000 \
-g=
```
* 测试
```
$ python tools/test.py
$ python tools/test.py configs/i3d-3x3-nl_r3d50_ucf101_rgb_224x32_dense.yaml outputs/i3d-3x3-nl_r3d50_ucf101_rgb_224x32_dense.pth
```
## 主要维护人员
* zhujian - *Initial work* - [zjykzj](https://github.com/zjykzj)
## 致谢
* [ facebookresearch/SlowFast](https://github.com/facebookresearch/SlowFast)
* [open-mmlab/mmaction2](https://github.com/open-mmlab/mmaction2)
```
@misc{fan2020pyslowfast,
author = {Haoqi Fan and Yanghao Li and Bo Xiong and Wan-Yen Lo and
Christoph Feichtenhofer},
title = {PySlowFast},
howpublished = {\url{https://github.com/facebookresearch/slowfast}},
year = {2020}
}
@misc{wang2018nonlocal,
title={Non-local Neural Networks},
author={Xiaolong Wang and Ross Girshick and Abhinav Gupta and Kaiming He},
year={2018},
eprint={1711.07971},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
```
## 参与贡献方式
欢迎任何人的参与！打开[issue](https://github.com/ZJCV/Non-local/issues)或提交合并请求。
注意:
* `GIT`提交，请遵守[Conventional Commits](https://www.conventionalcommits.org/en/v1.0.0-beta.4/)规范
* 语义版本化，请遵守[Semantic Versioning 2.0.0](https://semver.org)规范
* `README`编写，请遵守[standard-readme](https://github.com/RichardLitt/standard-readme)规范
## 许可证
[Apache License 2.0](LICENSE) © 2020 zjykzj