0x00
这里使用FCHD-Fully-Convolutional-Head-Detector人头识别的开源项目
系统:AWS Ubuntu Server 18.04 LTS(HVM)
类型:p2类型 (我使用的是p2.xlarge)
0x01 环境安装
首先更新系统,更新完建议重启
sudo apt update sudo apt upgrade -y
安装所需要的工具
sudo apt install gcc make python3-pip unzip -y
安装CUDA
wget "https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux" -O cuda.run sudo bash cuda.run
安装选项使用默认就好
安装所需要的依赖库
sudo apt install python3-tk -y sudo apt install libsm-dev -y
命令行输入
ldconfig /usr/local/cuda-10.0/lib64/
通过pip安装所需工具
总共需要安装以下工具:
·torch
·torchvision
·cupy-cuda100
·scikit-image
·opencv-python
·torchnet
命令:###使用国内源会下载更快,这里使用清华的源 yuan=https://pypi.tuna.tsinghua.edu.cn/simple pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl pip3 install torchvision cupy-cuda100 scikit-image opencv-python torchnet -i $yuan
0x02 部署
通过S3储存桶下载并解压文件:
wget https://s3.cn-north-1.amazonaws.com.cn/neural-network-model/FCHD-Fully-Convolutional-Head-Detector.zip unzip FCHD-Fully-Convolutional-Head-Detector
- 下载一个视频放入
FCHD-Fully-Convolutional-Head-Detector/video
中 命令行输入:
python3 head_detection_demo.py --video_path video/视频文件名
若没有提示cuda错误即成功