天天看點

輕量級openpose人體姿态檢測

1.去github下載下傳輕量級openpose人體姿态檢測項目:https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch(溫馨提示:github網站打不開可到gitee網站去搜尋下載下傳)

輕量級openpose人體姿态檢測

2.下載下傳checkpoint_iter_370000.pth模型,去我的網盤下載下傳https://pan.baidu.com/s/1Qjwq7uFDFSt60MQBwJW7Rg密碼:jybl

輕量級openpose人體姿态檢測

3.打開這個項目,放入模型,放一點視訊素材測試

輕量級openpose人體姿态檢測

4.檢測運作的代碼:python human-pose/demo.py --checkpoint-path checkpoint_iter_370000.pth --video 5.mp4,運作結果圖檔顯示

python human-pose/demo.py --checkpoint-path checkpoint_iter_370000.pth --video 5.mp4
           
輕量級openpose人體姿态檢測

5.運作後顯示CPU占用60%,由此說明此項目是由cpu運作,現在我來調至GPU處理圖檔,找到處理圖檔的py檔案,加上代碼:

import torch
torch.set_num_threads(1)
           
輕量級openpose人體姿态檢測
輕量級openpose人體姿态檢測

6. 處理後顯示GPU運作了,cpu記憶體降到了百分之20多

輕量級openpose人體姿态檢測

7.速度快,可實時檢測,同時檢測準确率高,此項目堪稱完美,成功了請求兄弟們一鍵三連。

繼續閱讀