现在,经过谷歌官方的优化,即使是硬件规格更低的Coral开发板也能流畅使用了,帧率甚至不输iPhone。
安装与使用
BodyPix依赖于最新版本的Coral API,而对于Coral开发板,则需要安装最新的Mendel OS。然后运行项目中的install_requirements.sh来安装其他依赖项。
python3 bodypix.py
以上命令还可以加入不同选项进行更细化的操作,比如加入—width和—height来控制输入图像的分辨率,这样就可以在运行的速度和精度之间进行取舍。
python3 bodypix.py --width
480
--height
360
# fast but low res
python3 bodypix.py --width
640
--height
480
# default
python3 bodypix.py --width
1280
--height
720
# slower but high res
除了更改分辨率外,项目中还包含MobileNet和ResNet两种模型供用户选择。
如果你面对显示器,想制造镜像的效果,则可以加入参数:
python3 bodypix.py --mirror
另外还能对输入帧的编码格式进行定义:
python3 bodypix.py --h264
python3 bodypix.py --jpeg
还有我们开头提到的匿名模式,加入参数—anonymize,就可以不显示人物的身体特征和骨架,仅仅显示外部轮廓。
python3 bodypix.py --anonymize
关于谷歌Coral
谷歌在2019年初在TensorFlow开发者大会上推出的一系列边缘AI设备,不仅有类似于树莓派的开发板,还有AI加速计算棒。它们都内置了边缘TPU,比树莓派更适合做AI运算。