0x00#

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

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

0x01 环境安装#

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

   1
   sudo apt update
   2
   sudo apt upgrade -y

2. 安装所需要的工具

   1
   sudo apt install gcc make python3-pip unzip -y

3. 安装CUDA

   1
   wget "https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux" -O cuda.run
   2
   sudo bash cuda.run

   安装选项使用默认就好
4. 安装所需要的依赖库

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

5. 命令行输入

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

6. 通过pip安装所需工具 总共需要安装以下工具： ·torch ·torchvision ·cupy-cuda100 ·scikit-image ·opencv-python ·torchnet 命令：

   1
   ###使用国内源会下载更快，这里使用清华的源
   2
   yuan=https://pypi.tuna.tsinghua.edu.cn/simple
   3
   
   4
   pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
   5
   pip3 install torchvision cupy-cuda100 scikit-image opencv-python torchnet -i $yuan

0x02 部署#

1. 通过S3储存桶下载并解压文件：

   1
   wget https://s3.cn-north-1.amazonaws.com.cn/neural-network-model/FCHD-Fully-Convolutional-Head-Detector.zip
   2
   unzip FCHD-Fully-Convolutional-Head-Detector

2. 下载一个视频放入FCHD-Fully-Convolutional-Head-Detector/video中
3. 命令行输入：

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

   若没有提示cuda错误即成功