0x00

这里使用FCHD-Fully-Convolutional-Head-Detector人头识别的开源项目

系统：AWS Ubuntu Server 18.04 LTS(HVM)
类型：p2类型 (我使用的是p2.xlarge)

0x01 环境安装

1. 首先更新系统,更新完建议重启

   sudo apt update
   sudo apt upgrade -y

2. 安装所需要的工具

   sudo apt install gcc make python3-pip unzip -y

3. 安装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

   安装选项使用默认就好

4. 安装所需要的依赖库

   sudo apt install python3-tk -y
   sudo apt install libsm-dev -y

5. 命令行输入

   ldconfig /usr/local/cuda-10.0/lib64/

6. 通过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 部署

1. 通过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

2. 下载一个视频放入FCHD-Fully-Convolutional-Head-Detector/video中

3. 命令行输入：

   python3 head_detection_demo.py  --video_path video/视频文件名

   若没有提示cuda错误即成功