配置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

通过配置环境真的能学到很多,遇到什么问题解决什么问题这个过程会让你不断成长,当然配置环境如果遇到各种问题也真的很考验心态,感觉别人半天解决的问题,自己却小好几天都没能完成(别人快也是因为之前的积累嘛),不要看了别人的介绍就觉得一定会成功,等你去实践的时候也许会遇到其他的坑,遇到了跳出来就是很大的进步,希望大家不要放弃,共同分享学习交流!

评论 42
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值