搞深度學習這東西配環(huán)境真的難。總是在達成目標的路上遇到各種錯誤。

本文僅記錄配置環(huán)境中的心得。Spatial Temporal Graph Convolutional Networks for Skeleton Based Action Recognition

?論文解讀可參考：https://blog.csdn.net/qq_36893052/article/details/79860328

文章鏈接：https://arxiv.org/abs/1801.07455

Github 代碼：https://github.com/yysijie/st-gcn

第一步：你要從github網站上，下載這個包

第二步：配置環(huán)境

安裝pytorch參考我的這個博客：https://blog.csdn.net/Dongjiuqing/article/details/89417616

安裝openpose參考我這個博客：https://blog.csdn.net/Dongjiuqing/article/details/89006405

【注】：openpose要配置caffe和opencv，具體參考博客，這兩個必須要弄好。（所以說呢，如果你不需要可視化的openpose界面的話，你不需要看這些博客，也不需要配caffe openpose的環(huán)境，你直接下載個pytorch和一些必要的庫就可以開始訓練和測試了。）

然后FFmpeg和其他依賴項先按照人家給的方法去安裝。

第三部：安裝和下載模型

到這一步，我都沒什么問題，最尷尬的事情來了。從人家的網站上可以看到，這一步之后就是運行示例代碼了。然而我卡住了。報了一個很奇怪的錯誤：看圖：

RuntimeError：Errors in loading state_dict for Model:......

百思不得其解。我是直接運行的他的代碼：

bash tools/get_models.sh

?這一步卡住了，當時不知道哪里錯了。以為是之前的環(huán)境配的不對，還是別的什么原因。一頭霧水。最后索性直接從頭重新配環(huán)境，然而到了運行示例代碼的時候，還是這個錯誤。日了。。。于是擱置了一段時間。這個環(huán)境就丟在了一遍。過了一周，實在沒辦法，代碼還是要實現(xiàn) 。找了學長的幫助，學長說是模型的問題。于是從下面的百度云盤上下載了那個模型。然后替換掉原來的。這個問題神奇般的解決了！

然而報了下面這么一個錯誤：

ValueError: No way to determine width or height from video.

我靠！心態(tài)崩了啊，有沒有，心態(tài)崩了啊！一個環(huán)境拖了20天還沒配好，那個絕望啊。看到報的錯誤的時候，感覺自己的skvideo或者是ffmpeg有問題。后來查了查，好像確實是有問題。ffmpeg按照github上給的方法安裝的版本是1.4，而這個skvideo要求至少是2.8的版本。（一個網站上說的，不知道真的假的。）然后我把用命令安裝的ffmpeg卸載了，用源碼安裝一個4.1.3版本的。具體操作如下。

#卸載github上已安裝的版本 pip uninstall ffmpeg

從這個網站上下載了4.1.3版本：https://ffmpeg.org/download.html

解壓之后安裝，進入到該文件夾后：

./configure make make install

第四部：運行Demo

python3 main.py demo --openpose '/home/sues/Desktop/openpose/build' --video '/home/sues/Desktop/st-gcn-master/resource/media/skateboarding.mp4' ?--device 0

感人啊，有沒有。配了那么久的環(huán)境終于搞好了。

===================================================================================================================================分割線========================================================

補充一下，上面的都是為了演示demo才要配caffe和opencv，openpose，如果只想去訓練的話，什么都不需要，直接按照github上操作就ok了，環(huán)境就是torch0.4.0，然后該裝的裝一下就好了。配置文件修改一下就可以訓練了。一塊1080TI，大概1天左右訓練好一個數(shù)據(jù)集。

?

總結

以上是生活随笔為你收集整理的配置ST-GCN体会的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。