這個月先寫一篇吧,后面要復習數學考試了,可能到時候就忘了。今天寫一個比較有意思的東西,關於人體的分割與姿態估計。如下圖所示:
圖片選自mask rcnn的論文,這里由於時間的關系,就不多敘述技術細節了,網上有很多關於mask rcnn的博客,這里的keypoints是在mask rcnn上又添加了一個keypoints分支,總的模型結構圖就變成如下形式了
展示一下具體效果:
我建了一個repo介紹如何對視頻和圖片進行如上處理,具體可以參考https://github.com/FangYang970206/MaskRCNN-Keypoint-Demo,沒有GPU支持的同學可以通過google colab使用免費GPU進行試玩,鏈接如下https://colab.research.google.com/github/FangYang970206/MaskRCNN-Keypoint-Demo/blob/master/demo.ipynb
最后感謝chrispolo提供的pre-trained model和相關代碼。
好了,就說這么多了,我去復習了,玩得開心~