【Linux】在AWS上配置人头识别环境

April 25, 2019

2538 views

1564 words

#0x00
这里使用[FCHD-Fully-Convolutional-Head-Detector](https://github.com/aditya-vora/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错误即成功

Last modification：October 24, 2019