Windows10下的AlphaPose配置,人体摔倒姿态识别
(AlphaPose算法流程)
實(shí)驗(yàn)室主頁:www.mvig.org/
項(xiàng)目首頁:https://github.com/MVIG-SJTU/AlphaPose
環(huán)境:win10 + RTX2070s + Anaconda3
# 1. 創(chuàng)建alphapose虛擬環(huán)境. conda create -n alphapose python=3.6 -y conda activate alphapose# 2. 安裝pytorch(由于RTX2070s cuda必須為10.0+,conda自動(dòng)在虛擬環(huán)境安裝cuda10.0) conda install pytorch torchvision cudatoolkit=10.0 -c pytorch測試安裝是否成功
??
# 3.?下載項(xiàng)目(老版本) git clone -b pytorch https://github.com/MVIG-SJTU/AlphaPose.git# 4.?安裝依賴包(更改pytorch torchvision版本為安裝版本) pip install -r requirements.txtduang,?ntpath安裝出錯(cuò),alphapose的issue中看到有人說ntpath not necessary,可以直接刪掉,不用安裝。運(yùn)行video_demo.py時(shí)注釋import ntpath(15行),將ntpath替換為?'os.path' (66行)
也可以一樣一樣來,運(yùn)行demo時(shí)缺什么裝什么吧!
# 5.?安裝Pillow、opencv、scipy、matplotlib、tqdm、visdom # Pillow版本大于7.0.0,報(bào)錯(cuò)ImportError: cannot import name 'PILLOW_VERSION' pip install Pillow==6.2.2? pip install opencv-python pip install scipy pip install matplotlib pip install tqdm pip install visdom # 6. 下載模型文件: # yolov3-spp.weights 放入./models/yolo#?duc_se.pth?(2018/08/30) 放入./models/sppe # 7. 運(yùn)行demo.py、video_demo.py # 可在opt.py修改參數(shù),如batchsize、SP(單進(jìn)程,否則報(bào)錯(cuò)) python demo.py --indir ${img_directory} --outdir examples/respython video_demo.py --video ${path to video} --outdir examples/res --save_video運(yùn)行圖片結(jié)果:
視頻運(yùn)行結(jié)果:
基于上述人體姿態(tài)估計(jì)方法,我們參加了中國華錄杯—定向算法賽(人體摔倒姿態(tài)識別)賽道的比賽。由于該比賽沒有提供有監(jiān)督訓(xùn)練集,我們通過對定位的人體設(shè)定了相應(yīng)的規(guī)則(檢測框?qū)捀弑取㈥P(guān)鍵點(diǎn)相對位置等)對人體是否摔倒進(jìn)行了識別。
成功晉級(●'?'●)
點(diǎn)擊閱讀原文,下載人體定位&姿態(tài)估計(jì)模型文件
提取碼:f39n
長按二維碼關(guān)注我們
有趣的靈魂在等你
總結(jié)
以上是生活随笔為你收集整理的Windows10下的AlphaPose配置,人体摔倒姿态识别的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 疫情当前,宅家学习不无聊,AI视频课程资
- 下一篇: java信息管理系统总结_java实现科