【文件属性】：
文件名称：FCHD:人头检测网络 “FAST AND ACCURATE HEAD DETECTION IN CROWDED SCENES”
文件大小：1.56MB
文件格式：ZIP
更新时间：2021-05-18 16:38:58
Python
FCHD
本工程是官方工程的python3版本，重构了部分代码，并训练得到了一个效果不错的模型。
论文地址：
官方工程地址：
安装
Clone
git clone https://github.com/embracesource-cv-com/FCHD.git
安装依赖
cd FCHD
pip install -r requirements.txt
数据配置
在下载Brainwash数据集，caffe预训练模型，以及人头检测模型
将预训练模型vgg16_caffe.pth放到./checkpoints/pre_trained目录下
将检测模型checkpoint_best.pth放到./checkpoints目录下
将brainwash数据集放到/home/dataset/fchd_datas目录下
当然你可以自定义各个路径，然后在config.py中修改相应的配置
预测
在工程root目
【文件预览】：
FCHD-master
----trainer.py(4KB)
----.gitignore(1KB)
----requirements.txt(541B)
----data()
--------__init__.py(0B)
--------preprocess.py(2KB)
--------dataset.py(2KB)
----networks()
--------base_net.py(622B)
--------__init__.py(0B)
--------anchor_target_layer.py(3KB)
--------detector.py(3KB)
--------proposal_layer.py(2KB)
--------rpn.py(3KB)
----utils()
--------visualize.py(6KB)
--------tools.py(5KB)
--------losses.py(769B)
----examples()
--------result_3.png(609KB)
--------test_3.jpg(138KB)
--------result_2.png(375KB)
--------result_1.png(409KB)
--------test_1.jpg(46KB)
--------test_2.jpg(41KB)
----README.md(1KB)
----config.py(745B)
----predict.py(2KB)
----train.py(4KB)