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错误即成功

Last modification:October 24th, 2019 at 04:20 pm
如果觉得我的文章对你有用,请随意赞赏