从视频中获取发丝都能看见的人体步态轮廓图

学习前言

         最近在研究步态识别，复现了相关的论文算法：比较火的GaitSet等，也从网上下载了中科大开源的数据集CASIA-B进行训练，当我自己制作了视频数据集之后，却不知道怎么转换为CASIA-B中一样的二值人体轮廓图，网上告知的是用实例分割办法，我尝试使用Mask-RCNN搭建了实例分割平台，并且进行了测试，最终的效果却非常的不尽人意：转换出来的二值轮廓图轮廓模糊不清晰，完全无法与05年的CASIA-B比较，更别说更新的数据集了。深度学习这个方法行不通之后，我把目光放在了机器学习，最后发现了一个非常棒的办法:背景减除法。

---

从视频中获取发丝都能看见的人体步态轮廓图

学习前言

我所做的工作

实现代码

---

我所做的工作

在这过程中，我使用了以下方法进行了实验：

· ❥ Mask-RCNN实例分割、

· ❥ BackgroundSubtractorCNT、

· ❥ BackgroundSubtractorGMG、

· ❥ BackgroundSubtractorGSOC、

· ❥ BackgroundSubtractorLSBP、

· ❥ BackgroundSubtractorMOG、

· ❥ BackgroundSubtractorMOG2、

· ❥ BackgroundSubtractorKNN

最终实验结果表明，对于人体轮廓图的提取，表现效果最好的是BackgroundSubtractorMOG。

先附上一张效果动态图：

        可以看到，效果非常的nice!实际上如果是画中的人头发够飘逸，我们甚至可以看到发丝的飘动！ 警告：拍摄的摄像头不可以动！具体原因是因为背景减除法的原理就是把不动的东西删掉！

实现代码

        大家看到这里，会不会觉得如果效果和博主说的一样好的话，那代码是不是非常复杂？其实不用，在此感谢方便的OpenCv，只需要简单的一个调用。就可以实现我上述的功能！

提醒一句：上述的Opencv是opencv_contrib!

        由于机器学习的这个代码有个缺陷就是前面几张乃至十几张效果有时候会非常不好，所以我们只选用中间30%到70%的质量比较高的图片。（当然这个范围也可以手动调整）

import os.path
import cv2
# ---------------------------

def from_video_get_img(video_path):
    print(video_path)
    print(video_path + ' Is Loading...')

    # 创建存放结果图片的文件
    save_file_name = video_path.split('/')[-1].split('.')[0]
    save_fgmask_path = 'Package/Fgmask/' + save_file_name + '/'
    if not os.path.exists(save_fgmask_path):
        os.makedirs(save_fgmask_path)

    # 读取视频
    cap = cv2.VideoCapture(video_path)
    if cap.isOpened():
        # best>MOG
        fgbg = cv2.bgsegm.createBackgroundSubtractorMOG()
        fgmask_img_list = []
        id = 0
        flag = True
        while flag:
            ret, frame = cap.read()
            if ret:
                fgmask = fgbg.apply(frame)

                fgmask_img_list.append(fgmask)

                id += 1
                cv2.imshow('frame', fgmask)
                cv2.waitKey(1)
            else:
                break
    cap.release()
    cv2.destroyAllWindows()
    # 写入中间30%-70%的图片张以确保质量
    length = len(fgmask_img_list)
    for i in range(int(length*0.3), int(length*0.7)+1):
        cv2.imwrite(save_fgmask_path+str(i)+'.png', fgmask_img_list[i])

if __name__ == "__main__":
    # 在此强调 video 一定要改名为：名字代号_走路状态_序号.视频类型 的格式，不然程序会报错
    # 如果想要不报错 请修改 save_file_name 这个变量的赋值过程
    from_video_get_img('video/2/tds_nm_03.MOV')

完毕！ 

 最后如果觉得我的文章有帮助，麻烦大家点赞一下，谢谢！整体项目可以看这篇博客：https://blog.csdn.net/qq_51831335/article/details/124319777?spm=1001.2014.3001.5502