配置ST-GCN环境记录【Google colab】
基于Google colab配置ST-GCN环境记录
前言:Google colab为Deep Learning Applications提供了一个免费的服务器平台,虽然存在很多限制,但是对于学习深度学习而缺乏设备的同学来说简直就是雪中送炭【Google colab使用方法自行学习】
最近在学习<Spatial Temporal Graph Convolutional Networks for Skeleton Based Action Recognition>。因此本文记录了在Google colab平台上配置ST-GCN环境的详细过程以及踩过的坑
文章链接:https://arxiv.org/abs/1801.07455
Github 代码:https://github.com/yysijie/st-gcn
一、从github上下载程序包并上传到谷歌云盘
二、配置环境
环境要求 - Python3 (>3.5) - [PyTorch](http://pytorch.org/) - [Openpose](https://github.com/CMU-Perceptual-Computing-Lab/openpose) **with** [Python API](https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/doc/installation.md#python-api). (Optional: for demo only) - Other Python libraries can be installed by `pip install -r requirements.txt` <!-- - FFmpeg (Optional: for demo only), which can be installed by `sudo apt-get install ffmpeg` -->
- Google colab默认使用python3.6.9
查看python版本(python3.6.9符合环境要求)
!python --version
- Google colab默认使用cuda10.0
本环境使用cuda9.0,安装如下
!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb # 安装软件包:dpkg -i <.deb file name>
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub # 下载的文件添加到本地数据库
!apt-get update
!apt-get install cuda=9.0.176-1
查看cuda版本
!cat /usr/local/cuda/version.txt
- Google colab默认使用pytorch1.5.1
该网络是在pytorch1.0.0基础上训练的,因此需要安装pytorch1.0.0(网上也有要求安装pytorch0.4.0版本,根据要求安装不同版本)
!pip install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
查看pytorch版本
import torch
print(torch.__version__)
安装指定版本的torchvision
!pip install torchvision==0.2.0
以上是pytorch与torchvision对应版本的一个参考,我安装了pytorch1.0.0与torchvision0.2.0
- 安装openpsoe
Google colab上部署和使用openpose可参考该博客:https://blog.csdn.net/yrwang_xd/article/details/103150691
- 安装环境所需的其他python库
!pip install -r requirements.txt
- 安装FFmpeg
!sudo apt-get install ffmpeg
三、安装和下载模型
- Installation
%cd torchlight
!python setup.py install
%cd ..
Google colab进入指定文件夹
import os
os.chdir("drive/My Drive/st-gcn-master") # ()中为具体路径
- Get pretrained models
bash tools/get_models.sh
获取的模型放在./models路径下。也可以通过谷歌云盘或者百度网盘获取模型文件
谷歌云盘:https://drive.google.com/drive/folders/1IYKoSrjeI3yYJ9bO0_z_eDo92i7ob_aF
百度网盘:https://pan.baidu.com/s/1dwKG2TLvG-R1qeIiE4MjeA#list/path=%2FShare%2FAAAI18%2Fst-gcn%2Fmodels&parentPath=%2FShare
以上是配置ST-CGN环境的过程,接下来就可以运行示例代码了,运行哪有那么容易成功0.0第一个问题如下
AttributeError: module 'yaml' has no attribute 'FullLoader'AttributeError: module 'yaml' has no attribute 'FullLoader'
答: FullLoader:该类仅在PyyAML 5.1及更高版本中可用.
查看PyyAML版本
import yaml
yaml.__version__
Google colab中PyyAML默认版本为3.1.1,因此需要更新PyyAML版本
!pip install -U PyYAML
解决了上面的小问题后出现了下面的问题(根据我上面的步骤走下来不会看到下面的报错)
答: 训练模型和测试加载模型所使用的环境不一致
这个问题困扰了我很久,网上有的解释到是因为Get pretrained models时使用运行代码获取的不对,换成通过谷歌云盘或者百度网盘下载的就解决了(可是我就是通过谷歌云盘直接下载的模型,还是报错0.0)
在配置这个环境时,参考网上博客说要安装pytorch0.4.0版本,后来我换成了pytorch1.0.0版本,通过这个方法解决了0.0(上面对pytorch版本没有具体要求,这是个坑啊0.0.,可能人家的训练模型就是用了pytorch0.4.0)
最后在运行示例代码时使用了绝对路径报错
!python main.py demo --video 绝对路径 --openpose 绝对路径
Can not find Openpose Python API.
看了看代码
openpose = '{}/examples/openpose/openpose.bin'.format(self.arg.openpose)
parser.add_argument('--video',default='./resource/media/skateboarding.mp4',
改成相对路径解决了0.0(心累)
额。。。看了上面的过程感觉我配的很轻松?no!no!no!可是足足愁了我一个星期,当然不止上面的问题,比如:
一直处于运行阶段,无法输出结果
这个应该就是Google colab这个服务器平台的问题了,出现这个问题之后重启尝试了4-5遍都不能正常输出,但是第二天重新打开之后这个问题就消失了0.0
通过配置环境真的能学到很多,遇到什么问题解决什么问题这个过程会让你不断成长,当然配置环境如果遇到各种问题也真的很考验心态,感觉别人半天解决的问题,自己却小好几天都没能完成(别人快也是因为之前的积累嘛),不要看了别人的介绍就觉得一定会成功,等你去实践的时候也许会遇到其他的坑,遇到了跳出来就是很大的进步,希望大家不要放弃,共同分享学习交流!
配置ST-GCN环境记录【Google colab】相关推荐
- 安装 Win10 Ubuntu 16.04 双系统以及 Ubuntu 配置深度学习环境记录
0. 前言 坑爹的Ubuntu晚上运行还是好好的,第二天中午的时候打开机器发现屏幕分辨率不正常了:2K屏显示800*600左右的分辨率(无法调节),一个图标一拳头大,窗口和网页显示不全.Google查 ...
- Google Colab 使用
文章目录 关于 Google Colab 使用示例 设置GPU环境 关于 Google Colab 官方说明:https://colab.research.google.com 推荐教程:机器君在学习 ...
- 如何在google colab创建/配置torch 0.3.1环境
如何在google colab创建/配置torch 0.3.1环境 背景 复现GAN论文,环境配置需要torch 0.3,但是我们的colab是最新的环境,这会导致某些预训练的包导入出错,下面实现怎么 ...
- 神经网络学习小记录69——Pytorch 使用Google Colab进行深度学习
神经网络学习小记录69--Pytorch 使用Google Colab进行深度学习 注意事项 学习前言 什么是Google Colab 相关链接 利用Colab进行训练 一.数据集与预训练权重的上传 ...
- [ 深度学习平台 ] —— Google Colab : 开源GPU的配置与使用
一.从0到1 1. 挂载云盘 ### 挂载云盘 from google.colab import drive drive.mount('/content/gdrive') 2. 在notebook中使 ...
- vscode中装js解释器_h5学习记录(1)--vscode配置js开发环境
文笔不是很好,第一次写东西,主要为了记录h5的学习过程.今天记录的是vscode配置js开发环境. 什么是VSCode Visual Studio Code (简称VS Code/VSC) 是一款于2 ...
- Mac配置LNMP开发环境全记录 一看就懂系列
本文操作环境:macOS10.15系统.php7.4.5版,MacBook Air 2019电脑 Mac下Docker配置LNMP开发环境 前言: 1.Docker的标准用法是每个docker容器只提 ...
- Google Colab——用谷歌免费GPU跑你的深度学习代码
Google Colab简介 Google Colaboratory是谷歌开放的一款研究工具,主要用于机器学习的开发和研究.这款工具现在可以免费使用,但是不是永久免费暂时还不确定.Google Col ...
- Google Colab使用教程
简介 Google Colaboratory是谷歌开放的云服务平台,提供免费的CPU.GPU和TPU服务器. 目前深度学习在图像和文本上的应用越来越多,不断有新的模型.新的算法获得更好的效果,然而,一 ...
最新文章
- asp.NET自定义服务器控件内部细节系列教程四
- Ubuntu 卸载 Nvidia 驱动和安装最新驱动
- 【Qt】简单QT文本编辑器
- static_cast, dynamic_cast, const_cast学习和探讨
- matplotlib 柱状图 分组_Python数据分析与可视化之matplotlib可视化(三)
- HDFS API编程
- 如何选择系统服务器,如何选择服务器的操作系统
- 3 Machine-Level Representation 程序的机器级表示
- SourceOffSite使用方法
- 【python脚本系列】Midi数字化乐器接口
- 百分比换算十六进制透明度
- 素材网下载图片要 VIP ? 不存在的
- 简单计算 ( 山东科技大学第二届ACM校赛)
- leafnotification_Notification Service
- OS模块--批量修改文件名字(二)
- Tabby终端模拟器在Mac上的配置和使用
- 当前你所在的服务器更新维护尚未完成,我们将于01月16日09:00-13:00对所有服务器进行更...
- k8s 命令 重启_k8s常用命令
- php机器码领取,通过IP和机器码来限制用户领取奖励脚本
- 全国高校专业JSON
热门文章
- Java 模拟面试题
- 记录一次ncl从前端到后端出图过程:调用linux-shell执行ncl命令从nc文件出图
- 正大集团oa系统服务器,正大oa服务器地址
- 运维(32) Prometheus+Grafana监控SpringBoot
- 医学图像分割方法及卷积神经网络在医学图像分割上的应用
- Linux GccGcc-c++安装
- Ubuntu-拼音输入法安装
- 产品读书《你的团队需要一个会讲故事的人》
- python 波动率锥_推广 || 12个小时干货分享!上海财大期权实战特训课程(11月)...
- 如何利用EXCEL生成任意自由度任意显著因子的F分布表