win10 ST-GCN复现
win10 ST-GCN复现
下载源码
https://github.com/yysijie/st-gcn
安装requirements
1.anaconda切换python环境
命令为 conda activate 环境名,如:
conda activate pytorch1.4
2.安装requirements
cd /d F:
cd st-gcn //切换到st-gcn目录下,win10可能要以管理员模式开启cmd
cd torchlight
python setup.py install
cd ..
pip install -r requirements.txt
提示安装失败,于是一个一个安装packge(之前安装过pytorch)
conda install pyyaml
conda install argparse
conda install numpy
conda install h5py
conda install opencv-python
conda install imageio
conda install scikit-video
安装opencv-python时出现问题,于是自己在清华镜像源下载本地安装包(这里要和python的版本对应),我是python3.6下载的opencv_python-3.4.1.15-cp36-cp36m-win_amd64.whl
下载完成以后用pip install 安装包路径名安装opencv
pip install F:/opencv_python-3.4.1.15-cp36-cp36m-win_amd64.whl
下载数据集
这里使用的ntu-rgbd数据集, 百度云版本的可以在这里下载,https://github.com/Hrener/3D-Action-recognition
将Raw数据集处理为ST-GCN的输入
python tools/ntu_gendata.py --data_path <path to nturgbd+d_skeletons>
<path to nturgbd+d_skeletons>为Raw_skeleton/skeleton+D0-30000解压后的目录路径
python tools/ntu_gendata.py --data_path F:\skeleton+D0-30000
在st-gcn/data/目录下可找到处理后的数据
训练模型
训练前遇到的一些的bug
F:\st-gcn>python main.py recognition -c config/st_gcn/ntu-xsub/train.yaml
出现错误 RuntimeError: CUDA error: invalid device ordinal,发现源码中有4个gpu,将train.yaml中的device=[0,1,2,3]改为device=[0]
出现错误 BrokenPipeError: [Errno 32] Broken pipe,将processor.py下的函数load_data中的两个DataLoader()函数的参数num_workers设置为0,避免使用多线程读取数据
出现错误 cuda runtime error (2) : out of memory, 经过尝试,4G显存将batch_size设置为8时(在config下的train.yaml中设置),显存占用率刚好接近100%
中途暂停
由于电脑不能一直开着,训练一个epoch要接近一个小时的时间,源码中num_epoch设置为80,想一次训练完不太可能。读源码时发现代码有提供训练中途保存模型的功能,可以在train.yaml文件中进行如下修改
start_epoch = 0 //这里填写上次训练中断时的epoch,如果上次训练了5个epoch(0,1,2,3,4),那么再次训练的时候填5即可
save_interval = 1 //源码是时10个epoch保存一次模型参数weights,我设置为1
weights = model_path //这里填写最近一个epoch保存的model_weights.pt文件的路径,如果不加上这个参数,则默认训练一个新的模型
每次训练开始时,修改start_epoch和weights以接着训练上次未完成的模型
win10 ST-GCN复现相关推荐
- Win10下快速复现Mask_RCNN避坑指南
硬件环境:笔记本电脑win10系统.1050Ti显卡 软件环境:Anaconda.Pycharm 项目地址:https://github.com/matterport/Mask_RCNN 一.环境搭建 ...
- Win10下手把手教你Mask R-CNN用自己的数据集训练(从labelme标记开始)
对于Mask R-CNN的环境配置不了解的同学可以看我之前的博客,详细的讲了如何复现跑通demo https://blog.csdn.net/hesongzefairy/article/details ...
- 论文阅读 Skeleton-based abnormal gait recognition with spatio-temporal attention enhanced
Skeleton-based abnormal gait recognition with spatio-temporal attention enhanced gait-structural gra ...
- Windows 内网渗透之委派攻击
文章目录 委派攻击 前言 非约束委派攻击 非约束委派大致流程 搭建复现 查询配置非约束委派 powerview 查询配置非约束委派 adfind查询非约束委派 非约束委派攻击案例 非约束委派& ...
- 域渗透委派攻击之约束委派
约束委派 约束委派(Constrained Delegation)即 Kerberos 的扩展协议 S4U2Proxy,服务账号只能获取某用户的 TGS ,从而只能模拟用户访问特定的服务,这也相对应非 ...
- msf win10漏洞_【CVE20200796】漏洞复现
一.漏洞简介 该漏洞是服务器消息块3.1.1(SMBv3)协议中处理压缩消息时,对数据没有经过安全检查,直接使用会引发内存破坏,成功利用针对此漏洞的攻击者可以获得在目标服务器或客户端上执行代码的能力. ...
- 【漏洞复现】CVE-2020-0796漏洞复现(win10系统)
前言: Microsoft Windows和Microsoft Windows Server都是美国微软(Microsoft)公司的产品,Microsoft Windows是一套个人设备使用的操作系统 ...
- GCN的几种模型复现笔记
引言 本篇笔记紧接上文,主要是上一篇看写了快2w字,再去接入代码感觉有点不太妙,后台都崩了好几次,因为内存不足,那就正好将内容分开来,可以水两篇,另外也给脑子放个假,最近事情有点多,思绪都有些乱,跳出 ...
- win10 永恒之黑复现
永恒之黑复现 一:漏洞简介 漏洞原理: SMB远程代码执行漏洞 SMB 3.1.1协议中处理压缩消息时,对其中数据没有经过安全检查,直接使用会引发内存破坏漏洞,可能被攻击者利用远程执行任意代码.攻击者 ...
最新文章
- Java中数据存储方式
- Unix/Linux环境C编程入门教程(39) shell命令之系统管理
- redis缓存数据表
- Linux管道和重定向 ---多命令协作
- copy a random link
- jvm面试 -- 谈谈ClassLoader ,ClassLoader的双亲委派机制 , loadClass和forName的区别
- 【SpringBoot】开启热部署(jsp 页面热部署、devtools 全局热部署)
- ansible一键部署rsync服务剧本
- java web 调度_javaweb车辆调度信息管理平台
- word2007显示控件工具箱
- 信息系统项目管理师必背核心考点(二十一)可交付成果
- 通过Safari浏览器获取iOS设备UDID(设备唯一标识符)
- 测相伪距观测方程、线性组合
- 美国国土安全部:Log4j 漏洞的影响将持续十年或更久
- WWW 2015 | LINE:大规模信息网络的嵌入
- python编程水上行走_普通人如何水上行走,只要速度足够的快,何愁不能水上漂...
- PHP环境安装到U盘,Windows To Go辅助工具一键把Win10系统安装到U盘 | 麦田一棵葱
- 墨者学院-SQL注入漏洞测试(报错盲注)
- 剧场版动画《巨虫列岛》1月上映决定!
- 男人不知道的十六件事