代码下载

下载代码，可以使用码云下载代码

git clone FSA-Net: [CVPR19] FSA-Net: Learning Fine-Grained Structure Aggregation for Head Pose Estimation from a Single Image

环境配置

1. 配置一个python3.6的环境

conda create tensorflow -n python=3.6

2. 安装一些依赖

conda install cudatoolkit=8.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn==7.1.3
pip install tensorflow==1.10.0 -i https://mirrors.aliyun.com/pypi/simple
pip install tensorflow-gpu==1.10.0 -i https://mirrors.aliyun.com/pypi/simple
pip install keras==2.2.0 -i https://mirrors.aliyun.com/pypi/simple

其他的依赖环境

pip install pandas -i https://mirrors.aliyun.com/pypi/simple
pip install Pillow -i https://mirrors.aliyun.com/pypi/simple
pip install sklearn -i https://mirrors.aliyun.com/pypi/simple
pip install matplotlib==3.1.1 -i https://mirrors.aliyun.com/pypi/simple
pip install opencv-python==4.1.1.26 -i https://mirrors.aliyun.com/pypi/simple
pip install tqdm
pip install moviepy==1.0.0 -i https://mirrors.aliyun.com/pypi/simple
pip install pydot
pip install mtcnn

可能还会遇到

问题1

OSError: `pydot` failed to call GraphViz.Please install GraphViz (https://www.graphviz.org/) and ensure that its executables are in the $PATH.

解决方法：

sudo apt install graphviz

问题2

DEBUG:root:Running training...Epoch 1/902021-09-07 17:27:13.491941: E tensorflow/stream_executor/cuda/cuda_dnn.cc:352] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERRORSegmentation fault (core dumped)

显存太小了，我换到服务器就可以了。

训练数据集处理

首先下载300W-LP数据集，

将300W-LP和AFLW2000放到data/type1/里面

将数据放在data/type1里面，

cd data/type1
sh run_created_db_type1.sh

针对BIWI数据集的处理，将Biwi Kinect Head Pose Database中的hpdb文件命名为BIWI，然后传到FSA-Net/data里面了，使用TYY_create_db_biwi.py处理数据。

训练和测试

训练
sh run_fsanet_train.sh

修改run_fsanet_train.sh（里面好多模型训练），随便选择一个模型去训练。

KERAS_BACKEND=tensorflow python FSANET_train.py --batch_size 16 --nb_epochs 90 --model_type 5 --db_name '300W_LP'

测试

sh run_fsanet_test.sh

问题

好像是不容易转成onnx模型，可以使用其他人使用pytorch代码实现的FSANet

如下：

https://github.com/omasaht/headpose-fsanet-pytorch

yydyz/headpose-fsanet-pytorch

ubuntu16.04下FSA-Net环境安装和训练相关推荐

Ubuntu16.04下ROS Kinetic的安装（2022）
文章目录 Ubuntu16.04下ROS Kinetic的安装(2022) 一.Ubuntu16.04配置二.ROS Kinetic安装过程 1.软件中心配置 2.ROS的下载安装 2.1添加源和密 ...
Ubuntu16.04下Mongodb官网安装部署步骤（图文详解）（博主推荐）
不多说,直接上干货! 在这篇博客里,我采用了非官网的安装步骤,来进行安装.走了弯路,同时,也是不建议.因为在大数据领域和实际生产里,还是要走正规的为好. Ubuntu16.04下Mongodb(离线安 ...
ubuntu16.04 Intel Realsense D435i 环境安装
ubuntu16.04 Intel D435i 开发背景 intel Realsense D435i 环境配置安装realsense D435i SDK以及realsense-ros 安装libr ...
Ubuntu16.04下双显卡电脑安装显卡驱动
Ubuntu16.04双显卡电脑安装nvidia显卡驱动前言博主的电脑是集显+独显840M,在ubuntu16.04下安装显卡驱动一直碰到一些问题,鉴于网上双显卡电脑安装nvidia显卡驱动的相关 ...
Ubuntu16.04下搭建开发环境及编译tiny4412 Android系统【转】
本文转载自:http://blog.csdn.net/songze_lee/article/details/72808631 版权声明:本文为博主原创文章,未经博主允许不得转载. 1.安装ssh服务器 ...
Ubuntu16.04下RealSense D435i的安装和使用
安装步骤一.安装SDK步骤二.ROS相关包安装三.运行测试包一.安装SDK步骤 RealSense D435i SDK安装 1.更新环境(PS:最后的dist-upgrade会更新你Ubunt ...
Ubuntu16.04下为Titan RTX 安装显卡驱动、CUDA、cudnn
1.禁用自带的 nouveau nvidia驱动 sudo vim /etc/modprobe.d/blacklist.conf 在文件夹最末尾处添加命令 blacklist nouveau opti ...
ubuntu16.04下gpu版caffe2安装
http://http://blog.csdn.net/meccaendless/article/de由于之前已经配置好了ubuntu16.04+cuda9.1+cudnn7.04环境,参考我的这篇博 ...
Ubuntu16.04下arm-linux-gcc交叉编译环境搭建
Ubuntu下arm-Linux-gcc交叉编译环境搭建参考:http://blog.csdn.net/hebbely/article/details/53992805 1.网上下载 arm-lin ...
Ubuntu16.04下NVIDIA显卡驱动安装（华硕飞行堡垒7代，1660ti）
Ubuntu下NVIDIA显卡驱动安装我的电脑配置,华硕飞行堡垒7代,1660ti 亲测有效,我的系统重装5次! 1.首先进行安装,可以参考一下链接安装 2.安装完毕后可能会出现这样的情况按住C ...

ubuntu16.04下FSA-Net环境安装和训练